saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
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
distributed [Distributed] DeleteKey API for c10d TCP Store (#45401) 2020-09-28 15:30:39 -07:00
experiments
ideep
image
mobile
mpi
observers
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
sgd [caffe2] Fix preprocessor checks for FMA 2020-10-11 11:48:32 -07:00
share
test
transforms
utils caffe2: use at::mt19937 instead of std::mt19937 (10x speedup) (#43987) 2020-10-16 16:08:35 -07:00
video
.clang-format
__init__.py
c2_aten_srcs.bzl
CMakeLists.txt [pytorch] rewrite of the python binding codegen with the v2 API (#46244) 2020-10-19 17:36:45 -07:00
README.md
release-notes.md
requirements.txt
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