mirror of https://github.com/llvm/torch-mlir
96d662647f
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. |
||
|---|---|---|
| .. | ||
| configs | ||
| debug | ||
| linalg_on_tensors_backends | ||
| stablehlo_backends | ||
| test_suite | ||
| tosa_backends | ||
| CMakeLists.txt | ||
| __init__.py | ||
| annotations.py | ||
| framework.py | ||
| registry.py | ||
| reporting.py | ||
| utils.py | ||