projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: f90153c) | patch
[tensorexpr] Add 'is_allocated' flag for buffers and use it to insert 'Alloc/Free...
authorHui Guo <huiguo@fb.com>
Wed, 8 Sep 2021 22:30:59 +0000 (15:30 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Wed, 8 Sep 2021 22:34:42 +0000 (15:34 -0700)
commit527348a6fe934dcfeeb88352e0cb9e263c76b78d
tree9c10d7f5363f723dfcffdcac687b500539bc6dcatree | snapshot
parentf90153cda30d3f9aaa74fb7af7566db51d380457commit | diff
[tensorexpr] Add 'is_allocated' flag for buffers and use it to insert 'Alloc/Free' stmts (#64226)

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/64226

Test Plan: Imported from OSS

Reviewed By: ZolotukhinM

Differential Revision: D30652221

Pulled By: huiguoo

fbshipit-source-id: ef9bb0e3db2c444b476e5fc23956bc34ae0f0111
torch/csrc/jit/tensorexpr/loopnest.cpp diff | blob | history
torch/csrc/jit/tensorexpr/loopnest.h diff | blob | history
torch/csrc/jit/tensorexpr/tensorexpr_init.cpp diff | blob | history
Domain: Machine Learning / ML Framework;