[TensorExpr] Make CacheReplacer and IndexFlattener mutate stmts/exprs inplace. (...
authorMikhail Zolotukhin <mvz@fb.com>
Thu, 19 Aug 2021 05:56:47 +0000 (22:56 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Thu, 19 Aug 2021 05:59:31 +0000 (22:59 -0700)
commit6e00b31b15ba9a09b6aa71b0da1ba200be482011
treebf7e7c8e83f69f1fd2285c188a985abbf9fb827a
parent1d62fb8a63a4c69f791f623e913ee1eabb56c344
[TensorExpr] Make CacheReplacer and IndexFlattener mutate stmts/exprs inplace. (#63527)

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

Test Plan: Imported from OSS

Reviewed By: navahgar

Differential Revision: D30411411

Pulled By: ZolotukhinM

fbshipit-source-id: efb14ee57b36537fa4fefa89bdd6bafe7151c012
test/cpp/tensorexpr/test_loopnest.cpp
torch/csrc/jit/tensorexpr/loopnest.cpp
torch/csrc/jit/tensorexpr/registerizer.cpp