torch-mlir/test/python
Ramiro Leal-Cavazos 96d662647f
Fix import of constant bool tensor parameters (#2047)
Bool tensors are represented in TorchScript as an array of
`int8_t`s. However, when importing them into Torch-MLIR, the importer
was assuming the array had `int32_t` elements, leading to the importer
reading into memory that was out of bounds. This commit fixes the
casting of the bool tensor.
2023-04-20 18:38:48 -07:00
..
importer/jit_ir Fix import of constant bool tensor parameters (#2047) 2023-04-20 18:38:48 -07:00
custom_op_shape_dtype_fn.py Add `extra_library` kwarg to `torch_mlir.compile` (#1986) 2023-03-30 09:20:19 -07:00
lit.local.cfg Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
smoketest.py WIP: No PyTorch dep (#1854) 2023-02-13 14:21:06 -06:00