frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

NNPACK: Use new bindings and custom thread pool

Browse Source 
Summary:
This change should dramatically (~10X) improve performance of convolution with NNPACK engine
Closes https://github.com/caffe2/caffe2/pull/1730

Reviewed By: sf-wind

Differential Revision: D6695895

Pulled By: Maratyszcza

fbshipit-source-id: 26291916811ef4cb819a59aec848c4e23668e568
This commit is contained in:
Marat Dukhan
2018-01-11 10:43:31 -08:00
committed by Facebook Github Bot
parent 0a8a18ca01
commit 224493d9ce
19 changed files with 43 additions and 98 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
caffe2/utils/threadpool/pthreadpool_impl.cc
View File

@ -18,7 +18,6 @@
#include "caffe2/utils/threadpool/pthreadpool_impl.h"
#include "caffe2/utils/threadpool/ThreadPool.h"
#if CAFFE2_THREADPOOL_MOBILE
//
// External API
@ -38,5 +37,3 @@ void pthreadpool_compute_1d(struct pthreadpool* threadpool,
size_t pthreadpool_get_threads_count(struct pthreadpool* threadpool) {
return threadpool->pool_->getNumThreads();
}
#endif // CAFFE2_THREADPOOL_MOBILE