Test int64 dtype (#3728)
Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/3728 Using the `I64toI32()` export pass, we now support i64 input and constants (tensors defined in the `nn.Module`) and we test the following cases: - `torch.randint()` i64 input - `torch.tensor()` i64 input / constants - `torch.arange()` i64 input / constants bypass-github-export-checks bypass-github-pytorch-ci-checks bypass-github-executorch-ci-checks ghstack-source-id: 229458711 Reviewed By: SS-JIA Differential Revision: D57649649 fbshipit-source-id: 3b69d456ebdce1f9c2f855614a4723feb469ae62
J
Jorge Pineda committed
bd3acf25b1f596266b8b9ede5beba6dbb864d494
Parent: 9068baf
Committed by Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
on 6/8/2024, 2:15:32 AM