summaryrefslogtreecommitdiff
path: root/cpu_ref
diff options
context:
space:
mode:
authorMiao Wang <miaowang@google.com>2017-07-11 17:14:19 +0000
committerandroid-build-merger <android-build-merger@google.com>2017-07-11 17:14:19 +0000
commitfbea329269ab3ee4abf8c4bc4e87fd9e402886b4 (patch)
treecd0be2b0366c4ca45ef9f03371f7e369b5fba3d8 /cpu_ref
parenteea6c2734f784efb59f094d511565ef69cecb441 (diff)
parentf107b0ce6d5fa3e4f0d8f78f8ecf0c8547e836ee (diff)
downloadrs-fbea329269ab3ee4abf8c4bc4e87fd9e402886b4.tar.gz
Merge "Make gemmlowp able to run with multi-threads."
am: f107b0ce6d Change-Id: I570acc44afadbe68ef2dcfe2cd2b73ce0a36b7a3
Diffstat (limited to 'cpu_ref')
-rw-r--r--cpu_ref/rsCpuIntrinsicBLAS.cpp4
1 files changed, 4 insertions, 0 deletions
diff --git a/cpu_ref/rsCpuIntrinsicBLAS.cpp b/cpu_ref/rsCpuIntrinsicBLAS.cpp
index 4b086341..d60a3b90 100644
--- a/cpu_ref/rsCpuIntrinsicBLAS.cpp
+++ b/cpu_ref/rsCpuIntrinsicBLAS.cpp
@@ -877,6 +877,10 @@ void RsdCpuScriptIntrinsicBLAS::kernelBNNM(size_t m, size_t n, size_t k,
#endif
// Using gemmlowp to calculate the low precision 8 bit GEMM.
+ // Set MaxNumThreads to 0. The value 0 lets the implementation query
+ // the system to determine the number of hardware threads
+ gemmlowp::eight_bit_int_gemm::SetMaxNumThreads(0);
+
bool transpose_a = true;
bool transpose_b = false;
bool transpose_c = true;