pytorch/caffe2/utils
David Reiss ad8c0e57ef Add a command-line flag for overriding pthreadpool size (#46781)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/46781

Test Plan: Passed it to speed_benchmark_torch and saw perf change.

Reviewed By: iseeyuan

Differential Revision: D24752889

Pulled By: dreiss

fbshipit-source-id: 762981510f271d20f76e33b6e6f361c4a6f48e6c
2020-11-05 21:30:54 -08:00
..
hip
math Optimize Scale function (#44913) 2020-09-18 14:31:33 -07:00
threadpool Add a command-line flag for overriding pthreadpool size (#46781) 2020-11-05 21:30:54 -08:00
bench_utils.cc
bench_utils.h
cast_test.cc
cast.h
cblas.h
CMakeLists.txt Fix BUILD_CAFFE2 if FBGEMM and NNPACK are not built (#45610) 2020-10-01 14:58:55 -07:00
conversions.h
cpu_neon.h
cpuid_test.cc
cpuid.cc
cpuid.h [caffe2] Use cpuinfo in perfkernels to simplify build dependency (#36371) 2020-04-10 13:26:34 -07:00
eigen_utils.h
fatal_signal_asan_no_sig_test.cc
filler.h
fixed_divisor_test.cc
fixed_divisor.h
GpuBitonicSort.cuh
GpuDefs.cuh CUDA RTX30 series support (#45489) 2020-09-29 18:19:23 -07:00
GpuScanUtils.cuh RIP CUDA <9.2: circleci, aten, and caffe2 (#36846) 2020-05-18 13:41:05 -07:00
map_utils.h
math_cpu.cc caffe2: use at::mt19937 instead of std::mt19937 (10x speedup) (#43987) 2020-10-16 16:08:35 -07:00
math_gpu_test.cc
math_gpu.cu RIP CUDA <9.2: circleci, aten, and caffe2 (#36846) 2020-05-18 13:41:05 -07:00
math_test.cc
math-detail.h
math.h
murmur_hash3.cc
murmur_hash3.h
proto_convert.cc
proto_convert.h
proto_utils_test.cc
proto_utils.cc [Onnxifi] Don't throw exception when we cannot write out debug files (#45979) 2020-10-08 00:18:24 -07:00
proto_utils.h [Onnxifi] Don't throw exception when we cannot write out debug files (#45979) 2020-10-08 00:18:24 -07:00
proto_wrap.cc
proto_wrap.h
signal_handler.cc
signal_handler.h
simple_queue_test.cc
simple_queue.h
smart_tensor_printer_test.cc
smart_tensor_printer.cc
smart_tensor_printer.h
string_utils.cc
string_utils.h Fix out-of-boundary access in caffe2::StartsWith (#36672) 2020-04-15 20:40:59 -07:00
zmq_helper.h