[nnc] Handled cast in index expression during inlining (#64716)
authorRaghavan Raman <raghavanr@fb.com>
Thu, 9 Sep 2021 15:26:16 +0000 (08:26 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Thu, 9 Sep 2021 15:30:52 +0000 (08:30 -0700)
commitb7c86365d11f6f9ec280c636e6abfec63b33104d
tree2aa64b0d995f49821cbbdb8aae0e9db498c47ab9
parent652a8bf7d0d104c21964076083684ad0936fe05e
[nnc] Handled cast in index expression during inlining (#64716)

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

Test Plan: Imported from OSS

Reviewed By: ZolotukhinM

Differential Revision: D30826388

Pulled By: navahgar

fbshipit-source-id: 7e446602f650527e0d954e437f0370602019e040
test/cpp/tensorexpr/test_loopnest.cpp
torch/csrc/jit/tensorexpr/eval.cpp
torch/csrc/jit/tensorexpr/eval.h
torch/csrc/jit/tensorexpr/loopnest.cpp