Define wc_store_fence for riscv (#1475)

[ROCm/rccl commit: c6c7b6db98]
Этот коммит содержится в:
Xeonacid authored and GitHub committed 2025-01-06 16:59:14 -08:00
1 родитель e61194239d
Коммит f3e883805c
1 file changed
+1 -1
+1 -1
Просмотреть файл
@@ -41,7 +41,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); }