OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
71cb13869b
pytorch/caffe2
History
Yifu Wang 6e1ba79b7f [re-land] Introduce 3 low-latency, intra-node allreduce algorithms for small messages to PyTorch (#114001) (#116125) 
This is an attempt to re-land https://github.com/pytorch/pytorch/pull/114001. The previous attempt used `std::array` in cuda kernels which wasn't compatible with Meta's internal build.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/116125
Approved by: https://github.com/yf225
2023-12-20 07:13:50 +00:00
..
contrib Revert "Add _foreach_clamp (#106574)" 2023-08-11 21:05:04 +00:00
core [ROCm] Disabling Kernel Asserts for ROCm by default - fix and clean up and refactoring (#114660) 2023-12-13 15:44:53 +00:00
cuda_rtc
db
distributed [4/N] Add -Wdeprecated and related fixes (#110204) 2023-10-07 19:46:08 +00:00
experiments
ideep [ONEDNN][BC-breaking] update onednn from v2.7.3 to v3.1.1 (#97957) 2023-08-25 12:13:18 +00:00
image
mobile
mpi [BE] Enforce missing override keyword (#104032) 2023-06-24 02:34:24 +00:00
observers
onnx
operators [Resubmit][S372460 follow up] Reduce embedding feature validation failure carry-on impact (#111838) 2023-10-28 03:50:33 +00:00
opt [codemod][lowrisk] Remove extra semi colon from caffe2/caffe2/opt/optimizer.cc (#115018) 2023-12-13 23:11:33 +00:00
perfkernels Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
predictor
proto extract torch.proto to its own library (#97614) 2023-03-30 10:35:03 +00:00
python Revert "[Reland2] Update NVTX to NVTX3 (#109843)" 2023-12-05 16:10:20 +00:00
quantization [BE]: Apply PYI autofixes to various types (#107521) 2023-08-20 02:42:21 +00:00
queue [caffe2] Replace CAFFE_ prefixes in static_tracepoint.h macros with TORCH_ (#106380) 2023-08-03 21:51:36 +00:00
serialize Reduce single reader check time for inline_container (#113328) 2023-11-09 22:02:28 +00:00
sgd
share Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
test
transforms [1/N] Enable Wunused-result and Wunused-variable in torch targets (#110722) 2023-10-08 23:43:45 +00:00
utils [ROCm] Fix caffe2 build with hipblasv2 api (#116073) 2023-12-20 04:02:29 +00:00
video
__init__.py
.clang-format
BUILD_MODE.bzl
CMakeLists.txt [re-land] Introduce 3 low-latency, intra-node allreduce algorithms for small messages to PyTorch (#114001) (#116125) 2023-12-20 07:13:50 +00:00
README.md
release-notes.md
requirements.txt
unexported_symbols.lds
VERSION_NUMBER
version_script.lds

README.md

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