summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMiao Wang <miaowang@google.com>2017-07-11 17:17:50 +0000
committerandroid-build-merger <android-build-merger@google.com>2017-07-11 17:17:50 +0000
commit701e5d0b8bba426ffad18c3bbf63a03194ec4fd9 (patch)
treecd0be2b0366c4ca45ef9f03371f7e369b5fba3d8
parenteea6c2734f784efb59f094d511565ef69cecb441 (diff)
parentfbea329269ab3ee4abf8c4bc4e87fd9e402886b4 (diff)
downloadrs-701e5d0b8bba426ffad18c3bbf63a03194ec4fd9.tar.gz
Merge "Make gemmlowp able to run with multi-threads." am: f107b0ce6d
am: fbea329269 Change-Id: I73323b0476500ebc47c7f975280f151201052592
-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;