llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/frontends/pytorch/test/ivalue_import
History
Sean Silva f49ebf1690 Add `!torch.int` type. 
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.
 		2021-06-17 07:28:23 -07:00
..
annotations Add `!torch.int` type. 2021-06-17 07:28:23 -07:00
README.md Rename tests to match the code they test 2021-02-25 13:31:33 -08:00
functions-that-call-methods.py Introduce native `!torch.none` type. 2021-06-14 13:30:58 -07:00
functions.py Introduce `!torch.tensor` / `!torch.vtensor` types. 2021-06-10 10:56:48 -07:00
list.py Add `!torch.int` type. 2021-06-17 07:28:23 -07:00
methods-derefine.py Add `!torch.int` type. 2021-06-17 07:28:23 -07:00
methods-locations.py Rename tests to match the code they test 2021-02-25 13:31:33 -08:00
methods.py Introduce `!torch.tensor` / `!torch.vtensor` types. 2021-06-10 10:56:48 -07:00
object-identity-error-submodule.py Rename tests to match the code they test 2021-02-25 13:31:33 -08:00
object-identity-error.py Rename tests to match the code they test 2021-02-25 13:31:33 -08:00
object-identity-torch-bug.py Add TorchList type and prim::ListConstruct #218 2021-06-10 14:31:35 -07:00
object-identity.py Introduce `!torch.tensor` / `!torch.vtensor` types. 2021-06-10 10:56:48 -07:00
prim.py Introduce native `!torch.none` type. 2021-06-14 13:30:58 -07:00
primitives.py Add `!torch.int` type. 2021-06-17 07:28:23 -07:00
quantization.py Add `!torch.int` type. 2021-06-17 07:28:23 -07:00
strings.py Add !torch.str type. 2021-06-15 10:10:59 -07:00
submodules-select.py Add `torch.prim.If` 2021-06-16 14:04:31 -07:00
submodules.py Add `!torch.int` type. 2021-06-17 07:28:23 -07:00
tensors.py Introduce `!torch.tensor` / `!torch.vtensor` types. 2021-06-10 10:56:48 -07:00
tuple.py Add `!torch.int` type. 2021-06-17 07:28:23 -07:00

README.md

ivalue_import

Most of the tests in this directory test importing of TorchScript torch::jit::Module's.

Modules are just one of many types of c10::IValue's and recursively contain c10::IValue's. Thus, the work of importing TorchScript modules is mainly about importing the wide variety of possible c10::IValue's, hence the name of this directory and the corresponding code in ivalue_importer.cpp that it exercises.