Define wc_store_fence for riscv (#1475)
[ROCm/rccl commit: c6c7b6db98]
Tá an tiomantas seo le fáil i:
1 tuismitheoir
e61194239d
tiomantas
f3e883805c
1 comhad athraithe
+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); }
|
||||
|
||||
Tagairt in Eagrán Nua
Cuir bac ar úsáideoir