pytorch/caffe2/utils
Junjie Bai ff608a9ff3 Back out "Revert D10123245: Back out "codemod cuda_gpu_id to device_id"" (#12232)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/12232

Original commit changeset: fca91fea58b7

This adds proper modifications to the DeviceType <->DeviceOption conversion code added in D10033396

Reviewed By: jerryzh168

Differential Revision: D10132473

fbshipit-source-id: 801ef777e2950982cb47b48051b1471a0a91e64b
2018-10-01 21:54:52 -07:00
..
flat_hash_map Implement calling of c10 ops from c2 (#9369) 2018-08-03 16:09:23 -07:00
hip Removing some dependency edges from Blob to other caffe2 (#12043) 2018-09-25 11:40:24 -07:00
threadpool Unify all *_EXPORT and *_IMPORT macros across c++ backend (#12019) 2018-09-25 17:41:05 -07:00
Array.cpp
Array.h Fix Array.h when compiled with C++17 (#11816) 2018-09-19 16:45:53 -07:00
Array_test.cpp Constexpr std::move / std::forward for C++11 (#11396) 2018-09-13 12:56:17 -07:00
bench_utils.cc build changes to make cpu unified build working. (#10504) 2018-08-15 17:22:36 -07:00
bench_utils.h build changes to make cpu unified build working. (#10504) 2018-08-15 17:22:36 -07:00
cast.h
cast_test.cc
cblas.h
CMakeLists.txt OperatorDef <==> NodeProto Conversion (#11621) 2018-09-19 08:41:33 -07:00
conversions.h Replace float16 with at::Half in caffe2 (#11676) 2018-09-20 18:55:17 -07:00
cpu_neon.h
cpuid.cc
cpuid.h
cpuid_test.cc
dummy.cpp
eigen_utils.h Update Im2Col related to make preparation for group conv in NHWC order. (#10439) 2018-08-15 17:10:24 -07:00
fatal_signal_asan_no_sig_test.cc Forward declarations of needed curand functions (#10911) 2018-09-05 16:56:26 -07:00
filler.h Remove many caffe2::TIndex and replace them with int64_t (#11943) 2018-09-22 18:11:04 -07:00
fixed_divisor.h
fixed_divisor_test.cc
GpuBitonicSort.cuh
GpuDefs.cuh Miscellaneous updates for CUDA 10 (#12017) 2018-09-24 11:58:32 -07:00
GpuScanUtils.cuh
map_utils.h
math-detail.h
math.h Add cuda version of SpatialBNOp also optimize SpatialBN on CPU (#10888) 2018-09-06 18:26:13 -07:00
math_cpu.cc Unify all *_EXPORT and *_IMPORT macros across c++ backend (#12019) 2018-09-25 17:41:05 -07:00
math_gpu.cu Remove many caffe2::TIndex and replace them with int64_t (#11943) 2018-09-22 18:11:04 -07:00
math_gpu_test.cc Removing some dependency edges from Blob to other caffe2 (#12043) 2018-09-25 11:40:24 -07:00
math_test.cc Remove many caffe2::TIndex and replace them with int64_t (#11943) 2018-09-22 18:11:04 -07:00
math_utils.cc Optimize reduce ops for 2d and 3d (#9992) 2018-08-04 13:53:58 -07:00
math_utils.h build changes to make cpu unified build working. (#10504) 2018-08-15 17:22:36 -07:00
Metaprogramming.cpp
Metaprogramming.h Add registerOperator overloads that infer the schema (#10048) 2018-08-03 11:45:49 -07:00
Metaprogramming_test.cpp
murmur_hash3.cc
murmur_hash3.h
Optional.cpp
Optional.h
proto_convert.cc New version of PT1 model format (#12149) 2018-10-01 15:57:02 -07:00
proto_convert.h New version of PT1 model format (#12149) 2018-10-01 15:57:02 -07:00
proto_utils.cc Back out "Revert D10123245: Back out "codemod cuda_gpu_id to device_id"" (#12232) 2018-10-01 21:54:52 -07:00
proto_utils.h Unify all *_EXPORT and *_IMPORT macros across c++ backend (#12019) 2018-09-25 17:41:05 -07:00
proto_utils_test.cc Back out "Revert D10123245: Back out "codemod cuda_gpu_id to device_id"" (#12232) 2018-10-01 21:54:52 -07:00
proto_wrap.cc New Serialization Proto 2018-09-11 10:55:43 -07:00
proto_wrap.h build changes to make cpu unified build working. (#10504) 2018-08-15 17:22:36 -07:00
signal_handler.cc Print blob sizes on fatal signal (#10766) 2018-08-23 13:39:55 -07:00
signal_handler.h More changes for hidden visibility (#10692) 2018-08-21 13:39:57 -07:00
simple_queue.h
simple_queue_test.cc
smart_tensor_printer.cc
smart_tensor_printer.h More changes for hidden visibility (#10692) 2018-08-21 13:39:57 -07:00
smart_tensor_printer_test.cc Remove many caffe2::TIndex and replace them with int64_t (#11943) 2018-09-22 18:11:04 -07:00
string_utils.cc
string_utils.h build changes to make cpu unified build working. (#10504) 2018-08-15 17:22:36 -07:00
thread_name.cc
thread_name.h build changes to make cpu unified build working. (#10504) 2018-08-15 17:22:36 -07:00
thread_pool.h build changes to make cpu unified build working. (#10504) 2018-08-15 17:22:36 -07:00
TypeList.cpp
TypeList.h Implement c10 ops needed for benchmark (#9360) 2018-08-03 16:09:27 -07:00
TypeList_test.cpp
TypeTraits.cpp
TypeTraits.h
TypeTraits_test.cpp
zmq_helper.h Unify all *_EXPORT and *_IMPORT macros across c++ backend (#12019) 2018-09-25 17:41:05 -07:00