pytorch/caffe2
cyy e64ddd1ab9 Upgrade NVTX to NVTX3 (#90689)
Due to recent upgrade to CUDA 11, we can upgrade NVTX to NVTX3 as well, which is a header only library that can simplify the building system a lot.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/90689
Approved by: https://github.com/soumith, https://github.com/malfet
2023-03-23 01:56:42 +00:00
..
contrib Follow up on CUDA 12 support for PyTorch/Caffe2 (#95582) 2023-02-27 04:39:56 +00:00
core [caffe2] dont call cudnnDestroy on thread exit (crashes on windows with cuda 11/12) (#95382) 2023-03-10 06:42:51 +00:00
cuda_rtc
db
distributed [BE] [1/3] Rewrite super() calls in caffe2 and benchmarks (#94587) 2023-02-11 18:19:48 +00:00
experiments [BE] Remove dependency on six and future (#94709) 2023-02-14 09:14:14 +00:00
ideep [MKLDNN] Rename pooling_avg to pooling_avg_exclude_padding (#90247) 2023-01-12 00:08:30 +00:00
image
mobile
mpi
observers
onnx
operators Fix some code issues (#92760) 2023-01-24 08:19:03 +00:00
opt some reference and move fixes (#95942) 2023-03-10 03:44:09 +00:00
perfkernels [BE] Enforce sign-compare (#96723) 2023-03-15 06:04:20 +00:00
predictor
proto move caffe2/proto/ to its own Bazel package (#97324) 2023-03-22 17:19:26 +00:00
python Upgrade NVTX to NVTX3 (#90689) 2023-03-23 01:56:42 +00:00
quantization
queue
serialize [BE] Enforce sign-compare (#96723) 2023-03-15 06:04:20 +00:00
sgd [CUDA] Drop CUDA 10 support (#89582) 2023-01-05 05:11:53 +00:00
share Fix some code issues (#92760) 2023-01-24 08:19:03 +00:00
test
transforms
utils remove dead proto_convert library (#97322) 2023-03-22 14:40:31 +00:00
video [codemod][llvm15] LLVM-15 fixes for caffe2/caffe2/video/video_decoder.cc (#89937) 2022-12-01 03:46:22 +00:00
.clang-format
__init__.py
BUILD_MODE.bzl
CMakeLists.txt Upgrade NVTX to NVTX3 (#90689) 2023-03-23 01:56:42 +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