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. |
||
---|---|---|
.. | ||
importer/jit_ir | ||
custom_op_shape_dtype_fn.py | ||
lit.local.cfg | ||
smoketest.py |