OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
cuda_stream_jit
pytorch/caffe2
History
Hao Lu 53dff784e2 [caffe2] Fix inplace ops in onnx::SsaRewrite (#46134) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/46134

Make sure in-place ops stay in-place after SsaRewrite. This seems to break the premise of SSA, but it's necessary to ensure correctness. Note here we only preserve the inplace ops that enforce inplace. Ops like `Relu` don't enforce inplace, they allow inplace.

(Note: this ignores all push blocking failures!)

Reviewed By: yinghai

Differential Revision: D24234957

fbshipit-source-id: 274bd3ad6227fce6a98e615aad7e57cd2696aec3
2020-10-22 13:26:31 -07:00
..
contrib Add Sigmoid operator from Caffe2 (#46286) 2020-10-15 16:13:37 -07:00
core [caffe2] Allow memonger to optimize nets with inplace(enforced) ops (#46560) 2020-10-22 13:23:33 -07:00
cuda_rtc
db Formatting cmake (to lowercase without space for if/elseif/else/endif) (#35521) 2020-03-27 14:25:17 -07:00
distributed [Distributed] DeleteKey API for c10d TCP Store (#45401) 2020-09-28 15:30:39 -07:00
experiments Remove __future__ imports for legacy Python2 supports (#45033) 2020-09-23 17:57:02 -07:00
ideep Ensure IDEEP transpose operator works correctly 2020-08-11 12:58:31 -07:00
image Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
mobile Make msg() and msg_with_backtrace() private (#37094) 2020-05-04 11:54:34 -07:00
mpi [codemod][caffe2] Run clang-format - 5/7 2020-06-30 15:45:11 -07:00
observers [codemod][caffe2] Run clang-format - 5/7 2020-06-30 15:45:11 -07:00
onnx [caffe2] Fix inplace ops in onnx::SsaRewrite (#46134) 2020-10-22 13:26:31 -07:00
operators [mlf][efficiency] add tensor inference function to last-n collector op (#46693) 2020-10-22 01:15:00 -07:00
opt [shape inference] add shape inference support 2020-10-15 00:46:06 -07:00
perfkernels [caffe2] Fix preprocessor checks for FMA 2020-10-11 11:48:32 -07:00
predictor caffe2: use at::mt19937 instead of std::mt19937 (10x speedup) (#43987) 2020-10-16 16:08:35 -07:00
proto [shape inference] adding a new flag to the struct 2020-10-09 19:29:08 -07:00
python Replace map(lambda constructs (#46462) 2020-10-22 09:50:22 -07:00
quantization [caffe2][dnnlowp] Remove openmp usage in quantize dnnlowp op 2020-10-20 19:33:56 -07:00
queue [caffe2] adds Cancel to SafeDequeueBlobsOp and SafeEnqueueBlobsOp (#45177) 2020-09-24 14:22:46 -07:00
serialize Enable TF32 support for cuDNN (#40737) 2020-09-01 15:34:24 -07:00
sgd [caffe2] Fix preprocessor checks for FMA 2020-10-11 11:48:32 -07:00
share Update pthreadpool to pthreadpool:029c88620802e1361ccf41d1970bd5b07fd6b7bb. (#40524) 2020-07-09 10:00:36 -07:00
test
transforms Replaced whitelisted with allowed (#41867) 2020-07-23 16:53:51 -07:00
utils caffe2: use at::mt19937 instead of std::mt19937 (10x speedup) (#43987) 2020-10-16 16:08:35 -07:00
video [ROCm][Caffe2] Enable MIOpen 3D Pooling (#38260) 2020-07-08 17:42:55 -07:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00
.clang-format
c2_aten_srcs.bzl Back out "Back out "Back out "Revert D18542342: Boxed variable dispatch""" (#30650) 2019-12-06 11:45:09 -08:00
CMakeLists.txt [pytorch] rewrite of the python binding codegen with the v2 API (#46244) 2020-10-19 17:36:45 -07:00
README.md Grammar patch 1 (.md) (#41599) 2020-07-20 10:19:08 -07:00
release-notes.md
requirements.txt Add requests as a legit dependency (#25596) 2019-09-04 17:43:37 -07:00
VERSION_NUMBER

README.md

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