pytorch/caffe2
cyy 5bbfb96203 [Reland] Upgrade NVTX to NVTX3 (#97582)
PR #90689 replaces NVTX with NVTX3. However, the torch::nvtoolsext is created only when the third party NVTX is used.
 This is clear a logical error. We now move the creation code out of the branch to cover all cases. This should fix the issues reported in the comments of  #90689.

It would be better to move configurations of the failed FRL jobs to CI tests so that we can find such issues early before merging.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/97582
Approved by: https://github.com/peterbell10
2023-08-14 16:55:25 +00:00
..
contrib Revert "Add _foreach_clamp (#106574)" 2023-08-11 21:05:04 +00:00
core [caffe2] Replace CAFFE_ prefixes in static_tracepoint.h macros with TORCH_ (#106380) 2023-08-03 21:51:36 +00:00
cuda_rtc
db
distributed [BE] Enforce missing override keyword (#104032) 2023-06-24 02:34:24 +00:00
experiments
ideep [BE] Enforce missing override keyword (#104032) 2023-06-24 02:34:24 +00:00
image
mobile
mpi [BE] Enforce missing override keyword (#104032) 2023-06-24 02:34:24 +00:00
observers
onnx
operators [caffe2] Add enforce inside ScatterAssignOp (#106882) 2023-08-10 21:46:13 +00:00
opt fix some typos (#106018) 2023-07-26 18:14:44 +00:00
perfkernels Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
predictor
proto
python Fix test_div in caffe2/caffe2/python:hypothesis_test (#106694) 2023-08-08 04:50:21 +00:00
quantization [BE] Enforce missing override keyword (#104032) 2023-06-24 02:34:24 +00:00
queue [caffe2] Replace CAFFE_ prefixes in static_tracepoint.h macros with TORCH_ (#106380) 2023-08-03 21:51:36 +00:00
serialize [TGIF][Inplace][Perf] Copy tensor to device with pinned memory & move copy weight sleep to getRecord (#106849) 2023-08-13 07:37:46 +00:00
sgd
share Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
test
transforms Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
utils [ROCm] use hipblas instead of rocblas (#105881) 2023-07-31 20:42:55 +00:00
video
.clang-format
__init__.py
BUILD_MODE.bzl
CMakeLists.txt [Reland] Upgrade NVTX to NVTX3 (#97582) 2023-08-14 16:55:25 +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