pytorch/caffe2
Yavuz Yetim a20c7ce848 Fix race in AtomicFetchAdd. (#13479)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/13479

Increases the lock scope to above Output() calls.

These calls potentially allocate the underlying blob/tensor
objects and multiple invocations race each other over the
same output blobs/tensors.

Reviewed By: bwasti

Differential Revision: D12891629

fbshipit-source-id: a6015cfdb08e352521a1f062eb9d94a971cfbdb0
2018-11-19 16:11:58 -08:00
..
contrib Use onnx proto_utils to support using protobuf-lite 2018-11-19 13:32:46 -08:00
core Tensor construction: combine Resize+mutable_data - 1/4 (#13942) 2018-11-19 15:33:50 -08:00
cuda_rtc Rename ndim() -> dim() - 1/6 2018-11-07 07:30:11 -08:00
db Change hip filename extension to .hip (#14036) 2018-11-16 11:55:59 -08:00
distributed Fix include paths for typeid.h (#13689) 2018-11-14 18:04:09 -08:00
experiments Rename ndim() -> dim() - 1/6 2018-11-07 07:30:11 -08:00
ideep Implementation copy operator for mkl-dnn (#12820) 2018-10-31 19:35:53 -07:00
image Windows shared build (#13550) 2018-11-16 12:16:28 -08:00
mobile Windows shared build (#13550) 2018-11-16 12:16:28 -08:00
mpi Fix include paths for typeid.h (#13689) 2018-11-14 18:04:09 -08:00
observers Revert D10428917: [Caffe2] Add cost into profile observer 2018-11-16 23:30:07 -08:00
onnx Change from using enum to int to store data_type 2018-11-17 09:24:03 -08:00
operators Fix race in AtomicFetchAdd. (#13479) 2018-11-19 16:11:58 -08:00
opt Use onnx proto_utils to support using protobuf-lite 2018-11-19 13:32:46 -08:00
perfkernels fix sparse_adagrad param_size overflow error (#14049) 2018-11-16 18:53:32 -08:00
predictor Windows shared build (#13550) 2018-11-16 12:16:28 -08:00
proto Store the optimize flag in module (#14166) 2018-11-19 14:34:05 -08:00
python Change hip filename extension to .hip (#14036) 2018-11-16 11:55:59 -08:00
quantization Add ultra low precision options (#14133) 2018-11-18 12:51:34 -08:00
queue Windows shared build (#13550) 2018-11-16 12:16:28 -08:00
serialize Windows shared build (#13550) 2018-11-16 12:16:28 -08:00
sgd Change hip filename extension to .hip (#14036) 2018-11-16 11:55:59 -08:00
share Windows shared build (#13550) 2018-11-16 12:16:28 -08:00
test move flags to c10 (#12144) 2018-10-04 02:09:56 -07:00
transforms Windows shared build (#13550) 2018-11-16 12:16:28 -08:00
utils Windows shared build (#13550) 2018-11-16 12:16:28 -08:00
video Remove calls to CopyFrom that can be sync (#13205) 2018-10-29 13:57:42 -07:00
__init__.py Experimental support for setup.py develop mode install 2018-02-12 23:36:18 -08:00
.clang-format Add clang-format style file to caffe2 2018-02-05 20:35:18 -08:00
CMakeLists.txt Move AT_CUDA_CHECK to c10 2018-11-19 08:20:10 -08:00
README.md Fix issues link in Caffe2 readme (#10711) 2018-08-20 16:55:11 -07:00
release-notes.md Cleanup of the shipit commit (#8956) 2018-06-27 14:41:59 -07:00
requirements.txt Add typing into caffe2 requirements.txt for USE_ATEN (#10047) 2018-07-30 20:09:21 -07:00
VERSION_NUMBER move flags to c10 (#12144) 2018-10-04 02:09:56 -07:00

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