OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 00:20:18 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Allow forcing single-threaded execution at runtime.

Browse Source 
Summary: Might be useful for the EXC_RESOURCE / CPU issues.

Reviewed By: salexspb

Differential Revision: D4565494

fbshipit-source-id: 74ac9edeba6334a46ee6799a93ca96eb68216439
This commit is contained in:
Andrew Tulloch 2017-02-15 19:22:59 -08:00 committed by Facebook Github Bot
parent 40534de705
commit 7d9a0a41fd
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
caffe2/utils/threadpool/ThreadPool.cc
View File

@ -1,6 +1,9 @@
#include "caffe2/utils/threadpool/ThreadPool.h"
#include "caffe2/core/logging.h"
CAFFE2_DEFINE_bool(caffe2_threadpool_force_inline, false,
"Force to always run jobs on the calling thread");
#if CAFFE2_THREADPOOL_MOBILE
namespace caffe2 {
@ -122,7 +125,8 @@ ThreadPool::run(const std::function<void(int, size_t)>& fn, size_t range) {
// If there are no worker threads, or if the range is too small (too
// little work), just run locally
bool runLocally = (threads_.empty() || range < minWorkSize_);
bool runLocally = threads_.empty() || range < minWorkSize_ ||
FLAGS_caffe2_threadpool_force_inline;
auto numThreads = threadInfo_.size();
size_t workUnitsPerThread = 0;