diff options
Diffstat (limited to 'internal/multi_thread_gemm.h')
-rw-r--r-- | internal/multi_thread_gemm.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/internal/multi_thread_gemm.h b/internal/multi_thread_gemm.h index 9223a42..0aacddb 100644 --- a/internal/multi_thread_gemm.h +++ b/internal/multi_thread_gemm.h @@ -73,6 +73,8 @@ inline void WriteBarrier() { asm volatile("dmb ishst" ::: "memory"); #elif defined(GEMMLOWP_X86) asm volatile("sfence" ::: "memory"); +#elif defined(__mips__) + MemoryBarrier(); #else #error "Unsupported architecture for WriteBarrier." #endif @@ -85,6 +87,8 @@ inline void ReadBarrier() { asm volatile("dmb ishld" ::: "memory"); #elif defined(GEMMLOWP_X86) asm volatile("lfence" ::: "memory"); +#elif defined(__mips__) + MemoryBarrier(); #else #error "Unsupported architecture for ReadBarrier." #endif |