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. |
||
---|---|---|
.. | ||
ivalue_import | ||
node_import | ||
get_registered_ops.py | ||
lit.local.cfg |