[nnc] Updated indices during broadcast to use int64_t (#64627)
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:29:37 +0000 (08:29 -0700)
commit652a8bf7d0d104c21964076083684ad0936fe05e
tree83dc415f5023a36d646f70026940aceb777da762
parent459653a0f68098821d2ad2f3ff03759b595b4e9f
[nnc] Updated indices during broadcast to use int64_t (#64627)

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

This fixes the root cause of S242719

Test Plan: Imported from OSS

Reviewed By: ZolotukhinM

Differential Revision: D30801686

Pulled By: navahgar

fbshipit-source-id: b6d3ebdc7eb57116eaced53c2f35c7798bb17e80
test/cpp/tensorexpr/test_kernel.cpp
test/test_tensorexpr.py
torch/csrc/jit/tensorexpr/kernel.cpp