mirror of https://github.com/llvm/torch-mlir
f49ebf1690
This replaces the ad-hoc use of `i64` throughout the Torch layer, and helps to keep it crystal clear the distinction between `!torch.int` (which is modeling the Python `int` type) and the various types that serve as dtypes of tensors, which are a totally different type universe. Changes: - `!torch.int` type and C bindings. - Change `torch.constant.int` parser to not need the `: i64` at the end. - `m_TorchConstantInt` matcher to aid with matching constants. - BackendTypeConversion changes for `!torch.int` -> `i64` type conversion. - Refactor finalizing patterns in FinalizingBackendTypeConversionPass (they were getting very repetitive). - Mechanical rewriting of `!torch.int` to `i64` in all the tests, and `AnyTorchIntType` to `Torch_IntType` in the `.td` files. |
||
---|---|---|
.. | ||
test_arange.py | ||
test_conv_nllloss_grads.py | ||
test_export_add3.py | ||
test_export_batchnorm.py | ||
test_export_cat.py | ||
test_export_conv2d_fwd.py | ||
test_export_multi_out.py |