pytorch/caffe2
Jeff Daily ae9a4fa63c [ROCm] enforce ROCM_VERSION >= 6.0 (#125646)
Remove any code relying on ROCM_VERSION < 6.0.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/125646
Approved by: https://github.com/albanD, https://github.com/eqy
2024-05-12 18:01:28 +00:00
..
core [ROCm] enforce ROCM_VERSION >= 6.0 (#125646) 2024-05-12 18:01:28 +00:00
cuda_rtc [ROCm] avoid heap buffer overflow in hiprtc failure logs (#121865) 2024-04-24 01:09:08 +00:00
onnx/torch_ops Remove unused caffe2 subdirs (#125818) 2024-05-09 22:57:55 +00:00
perfkernels [caffe2] Add an avx512 implementation of adagrad_update (#113289) 2024-02-15 01:45:30 +00:00
proto rename ort to maia (#123265) 2024-04-23 00:33:25 +00:00
python Remove caffe2 contrib and experiments (#125038) 2024-04-29 06:27:13 +00:00
quantization Remove unused caffe2 subdirs (#125818) 2024-05-09 22:57:55 +00:00
serialize Revert "Add write_record_metadata to PyTorchFileWriter (#125184)" 2024-05-05 22:40:00 +00:00
test
utils [ROCm] enforce ROCM_VERSION >= 6.0 (#125646) 2024-05-12 18:01:28 +00:00
__init__.py
.clang-format
BUILD_MODE.bzl
CMakeLists.txt Remove Caffe2 python (#125143) 2024-05-10 21:15:43 +00:00
README.md
release-notes.md
requirements.txt
unexported_symbols.lds
VERSION_NUMBER
version_script.lds

Caffe2

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