pytorch/caffe2
efiks ea0ec9d71c [tourch] BatchBoxCox - fix numerical issue in vectorized code (#88875)
Summary:
Usage of fast math in BatchBoxCox kernel provided different math results between dev and optimized versions which cause few internal test to fail.
For now disabling the compiler optimized version and relying on ATEN vectors

Differential Revision: D41211784

Pull Request resolved: https://github.com/pytorch/pytorch/pull/88875
Approved by: https://github.com/hyuen
2022-11-11 21:58:23 +00:00
..
contrib Update CMakeLists.txt (#87030) 2022-10-28 04:56:40 +00:00
core Introduce TORCH_DISABLE_GPU_ASSERTS (#84190) 2022-11-04 04:43:05 +00:00
cuda_rtc
db
distributed
experiments
ideep
image
mobile Fix typos under caffe2 directory (#87840) 2022-10-28 04:53:36 +00:00
mpi
observers
onnx
operators Check all CUDA API calls for errors in caffe2/ (#81816) 2022-10-28 00:41:06 +00:00
opt
perfkernels [tourch] BatchBoxCox - fix numerical issue in vectorized code (#88875) 2022-11-11 21:58:23 +00:00
predictor
proto
python extract out the clean workspace test to its own file (#88682) 2022-11-09 13:48:57 +00:00
quantization Fix typos under caffe2 directory (#87840) 2022-10-28 04:53:36 +00:00
queue
serialize Fix PyTorchStreamWriter exception handling (#88128) 2022-10-31 23:38:03 +00:00
sgd
share
test
transforms
utils Check all CUDA API calls for errors in caffe2/ (#81816) 2022-10-28 00:41:06 +00:00
video
.clang-format
__init__.py
BUILD_MODE.bzl
CMakeLists.txt Update gloo submodule (#88530) 2022-11-09 01:04:32 +00:00
README.md
release-notes.md Fix typos under caffe2 directory (#87840) 2022-10-28 04:53:36 +00:00
requirements.txt
unexported_symbols.lds
VERSION_NUMBER
version_script.lds

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