experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Automated rollback of commit 92511b345c 

PiperOrigin-RevId: 254261347
This commit is contained in:
Yifei Feng 2019-06-20 13:35:11 -07:00 committed by TensorFlower Gardener
parent b71a4c0765
commit aaa415996e
3 changed files with 7 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/core/common_runtime/local_device.cc
View File

@ -81,7 +81,13 @@ struct LocalDevice::EigenThreadPoolInfo {
intra_op_parallelism_threads = env_num_threads;
// If no session setting or environment, compute a reasonable default.
if (intra_op_parallelism_threads == 0) {
intra_op_parallelism_threads = port::MaxParallelism(numa_node);
intra_op_parallelism_threads = port::NumSchedulableCPUs();
if (numa_node != port::kNUMANoAffinity) {
// Assume that CPUs are equally distributed over available NUMA nodes.
// This may not be true, but there isn't currently a better way of
// determining the number of CPUs specific to the requested node.
intra_op_parallelism_threads /= port::NUMANumNodes();
}
}
}
ThreadOptions thread_opts;

5
tensorflow/core/platform/cpu_info.h
View File

@ -46,11 +46,6 @@ int NumSchedulableCPUs();
// called during initialization, i.e., before before main() has started.
int MaxParallelism();
// Returns an estimate for the maximum parallelism for this process on the
// provided numa node, or any numa node if `numa_node` is kNUMANoAffinity.
// See MaxParallelism() for more information.
int MaxParallelism(int numa_node);
// Returns the total number of CPUs on the system. This number should
// not change even if the underlying cluster management software may
// change the number of schedulable CPUs. Unlike `NumSchedulableCPUs`, if the

10
tensorflow/core/platform/posix/port.cc
View File

@ -82,16 +82,6 @@ int NumSchedulableCPUs() {
int MaxParallelism() { return NumSchedulableCPUs(); }
int MaxParallelism(int numa_node) {
if (numa_node != port::kNUMANoAffinity) {
// Assume that CPUs are equally distributed over available NUMA nodes.
// This may not be true, but there isn't currently a better way of
// determining the number of CPUs specific to the requested node.
return NumSchedulableCPUs() / port::NUMANumNodes();
}
return NumSchedulableCPUs();
}
int NumTotalCPUs() {
int count = absl::base_internal::NumCPUs();
return (count <= 0) ? kUnknownCPU : count;