diff options
Diffstat (limited to 'eight_bit_int_gemm/eight_bit_int_gemm.cc')
| -rw-r--r-- | eight_bit_int_gemm/eight_bit_int_gemm.cc | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/eight_bit_int_gemm/eight_bit_int_gemm.cc b/eight_bit_int_gemm/eight_bit_int_gemm.cc index 8113bf3..512c483 100644 --- a/eight_bit_int_gemm/eight_bit_int_gemm.cc +++ b/eight_bit_int_gemm/eight_bit_int_gemm.cc @@ -307,7 +307,7 @@ void EightBitIntGemm(bool transpose_a, bool transpose_b, bool transpose_c, std::int32_t b_offset, int ldb, std::uint8_t* c, std::int32_t c_offset, std::int32_t c_mult_int, std::int32_t c_shift, int ldc, BitDepthSetting bit_depth) { - AutoGlobalLock<EightBitIntGemmLockId> lock; + ScopedLock sl(GlobalMutexes::EightBitIntGemm()); GemmContext* context = GetOrCreateGlobalContext(); #if defined(GEMMLOWP_USE_META_FASTPATH) && defined(GEMMLOWP_NEON) @@ -344,7 +344,7 @@ void EightBitIntGemm(bool transpose_a, bool transpose_b, bool transpose_c, const std::uint8_t* b, std::int32_t b_offset, std::int32_t ldb, float* c, float c_offset, std::int32_t ldc, BitDepthSetting bit_depth) { - AutoGlobalLock<EightBitIntGemmLockId> lock; + ScopedLock sl(GlobalMutexes::EightBitIntGemm()); GemmContext* context = GetOrCreateGlobalContext(); #if defined(GEMMLOWP_USE_META_FASTPATH) && defined(GEMMLOWP_NEON) @@ -405,13 +405,13 @@ void EightBitIntGemm(bool transpose_a, bool transpose_b, bool transpose_c, } void SetMaxNumThreads(int n) { - AutoGlobalLock<EightBitIntGemmLockId> lock; + ScopedLock sl(GlobalMutexes::EightBitIntGemm()); GemmContext* context = GetOrCreateGlobalContext(); context->set_max_num_threads(n); } void FreePersistentResources() { - AutoGlobalLock<EightBitIntGemmLockId> lock; + ScopedLock sl(GlobalMutexes::EightBitIntGemm()); DestroyGlobalContext(); DestroyGlobalScratch(); } |