mirror of https://github.com/llvm/torch-mlir
290d7755fb
follow up #761: This patch updates the `torch_mlir::convertTensorToMlirElementsAttr()` method to enable the creation of tensors whose base type is Float16. This patch also adds a test to validate the IR generation, and it updates the test for importing tensors of various types. |
||
---|---|---|
.. | ||
CMakeLists.txt | ||
TorchUpstream.cpp | ||
Utils.cpp |