forked from mindspore-Ecosystem/mindspore
21 lines
944 B
Diff
21 lines
944 B
Diff
|
diff --git a/src/common/dnnl_thread.hpp b/src/common/dnnl_thread.hpp
|
||
|
|
index 342bc3b00..0b9190f9c 100644
|
||
|
|
--- a/src/common/dnnl_thread.hpp
|
||
|
|
+++ b/src/common/dnnl_thread.hpp
|
||
|
|
@@ -104,10 +104,11 @@ inline int dnnl_get_max_threads() {
|
||
|
|
def_max_threads
|
||
|
|
= (int)dnnl::impl::cpu::platform::get_max_threads_to_use();
|
||
|
|
assert(def_max_threads > 0);
|
||
|
|
- // Use the default value if the threadpool-provided is outside the range
|
||
|
|
- // [1, def_max_threads]
|
||
|
|
- return tp ? std::min(std::max(1, tp->get_num_threads()), def_max_threads)
|
||
|
|
- : def_max_threads;
|
||
|
|
+
|
||
|
|
+ // Make user responsible for number of threads provided at execution time.
|
||
|
|
+ // This relates to the fact that the library may identify `def_max_threads`
|
||
|
|
+ // incorrectly for a platform.
|
||
|
|
+ return tp ? std::max(1, tp->get_num_threads()) : def_max_threads;
|
||
|
|
}
|
||
|
|
inline int dnnl_in_parallel() {
|
||
|
|
using namespace dnnl::impl::threadpool_utils;
|