torch-mlir

History

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
..
main.py	mhlo: migrate conversion to stablehlo (#1840 )	2023-02-02 07:29:47 -06:00
xfail_sets.py	Fix import of constant bool tensor parameters (#2047 )	2023-04-20 18:38:48 -07:00

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

main.py

mhlo: migrate conversion to stablehlo (#1840 )

2023-02-02 07:29:47 -06:00

xfail_sets.py

Fix import of constant bool tensor parameters (#2047 )

2023-04-20 18:38:48 -07:00