From 0de8a4b1c1c28bafa6875569ed883deeef47c78d Mon Sep 17 00:00:00 2001
From: Xeonacid <h.dwwwwww@gmail.com>
Date: Mon, 19 Feb 2024 10:05:01 +0800
Subject: [PATCH] Add wc_store_fence support for RISC-V

---
 src/include/gdrwrap.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/include/gdrwrap.h b/src/include/gdrwrap.h
index a64674cc5..cf55638f5 100644
--- a/src/include/gdrwrap.h
+++ b/src/include/gdrwrap.h
@@ -40,7 +40,7 @@ static inline void wc_store_fence(void) { asm volatile("sync") ; }
 #elif defined(__x86_64__)
 #include <immintrin.h>
 static inline void wc_store_fence(void) { _mm_sfence(); }
-#elif defined(__aarch64__)
+#elif defined(__aarch64__) || defined(__riscv)
 #ifdef __cplusplus
 #include <atomic>
 static inline void wc_store_fence(void) { std::atomic_thread_fence(std::memory_order_release); }