pytorch/caffe2
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
..
contrib Back out "Revert D10123245: Back out "codemod cuda_gpu_id to device_id"" (#12232) 2018-10-01 21:54:52 -07:00
core Back out "Revert D10123245: Back out "codemod cuda_gpu_id to device_id"" (#12232) 2018-10-01 21:54:52 -07:00
cuda_rtc Remove many caffe2::TIndex and replace them with int64_t (#11943) 2018-09-22 18:11:04 -07:00
db Unify all *_EXPORT and *_IMPORT macros across c++ backend (#12019) 2018-09-25 17:41:05 -07:00
distributed Move blob serialization to free functions (#11817) 2018-09-20 23:27:34 -07:00
experiments Remove many caffe2::TIndex and replace them with int64_t (#11943) 2018-09-22 18:11:04 -07:00
ideep Back out "[pytorch][PR] Revert "Move CreateContext to global registry (#11688)"" (#12121) 2018-10-01 11:10:46 -07:00
image Remove many caffe2::TIndex and replace them with int64_t (#11943) 2018-09-22 18:11:04 -07:00
mkl Back out "Revert D10123245: Back out "codemod cuda_gpu_id to device_id"" (#12232) 2018-10-01 21:54:52 -07:00
mobile Back out "[caffe2][mpscnn] Enable multiple external output" 2018-10-01 16:55:32 -07:00
mpi Remove many caffe2::TIndex and replace them with int64_t (#11943) 2018-09-22 18:11:04 -07:00
observers Back out "Revert D10123245: Back out "codemod cuda_gpu_id to device_id"" (#12232) 2018-10-01 21:54:52 -07:00
onnx Back out "Revert D10123245: Back out "codemod cuda_gpu_id to device_id"" (#12232) 2018-10-01 21:54:52 -07:00
operators Back out "Revert D10123245: Back out "codemod cuda_gpu_id to device_id"" (#12232) 2018-10-01 21:54:52 -07:00
opt Add caffe2_api to exported functions (#12184) 2018-09-28 18:12:00 -07:00
perfkernels set up c10 scaffolding. Move macros proper first. 2018-09-24 11:09:59 -07:00
predictor Removing some dependency edges from Blob to other caffe2 (#12043) 2018-09-25 11:40:24 -07:00
proto Back out "Revert D10123245: Back out "codemod cuda_gpu_id to device_id"" (#12232) 2018-10-01 21:54:52 -07:00
python Back out "Revert D10123245: Back out "codemod cuda_gpu_id to device_id"" (#12232) 2018-10-01 21:54:52 -07:00
queue Unify all *_EXPORT and *_IMPORT macros across c++ backend (#12019) 2018-09-25 17:41:05 -07:00
sgd Fallback CreateMutex/AtomicIter operators for mkl-dnn 2018-09-25 17:41:08 -07:00
share Removing some dependency edges from Blob to other caffe2 (#12043) 2018-09-25 11:40:24 -07:00
test
transforms caffe2::DeviceType -> at::DeviceType (#11254) 2018-09-05 16:28:09 -07:00
utils Back out "Revert D10123245: Back out "codemod cuda_gpu_id to device_id"" (#12232) 2018-10-01 21:54:52 -07:00
video Remove many caffe2::TIndex and replace them with int64_t (#11943) 2018-09-22 18:11:04 -07:00
.clang-format
__init__.py
CMakeLists.txt Enable ROCM clang-7 build 2018-10-01 15:11:40 -07:00
README.md Fix issues link in Caffe2 readme (#10711) 2018-08-20 16:55:11 -07:00
release-notes.md
requirements.txt
VERSION_NUMBER

Caffe2

Jenkins Build Status

Caffe2 is a lightweight, modular, and scalable deep learning framework. Building on the original Caffe, Caffe2 is designed with expression, speed, and modularity in mind.

Questions and Feedback

Please use Github issues (https://github.com/pytorch/pytorch/issues) to ask questions, report bugs, and request new features.

Further Resources on Caffe2.ai