diff --git a/projects/rccl/src/include/gdrwrap.h b/projects/rccl/src/include/gdrwrap.h index a75831149f..cb060b1f78 100644 --- a/projects/rccl/src/include/gdrwrap.h +++ b/projects/rccl/src/include/gdrwrap.h @@ -41,7 +41,7 @@ static inline void wc_store_fence(void) { asm volatile("sync") ; } #elif defined(__x86_64__) #include static inline void wc_store_fence(void) { _mm_sfence(); } -#elif defined(__aarch64__) +#elif defined(__aarch64__) || defined(__riscv) #ifdef __cplusplus #include static inline void wc_store_fence(void) { std::atomic_thread_fence(std::memory_order_release); }