torch-mlir/lib/Dialect/Torch
Ashay Rane 809f240f01
importer: add initial support for loading BFloat16 tensors (#761)
This patch updates the `torch_mlir::convertTensorToMlirElementsAttr()`
method to enable the creation of tensors whose base type is BFloat16.
This patch also adds a test to validate the IR generation, and it
updates the test for importing tensors of various types.
2022-04-29 09:01:49 -07:00
..
IR Revert "[MLIR][TORCH] Add E2E support for aten.ge.float op" 2022-04-28 07:49:58 -07:00
Transforms [MLIR][TORCH] Add E2E support for aten.index_put.hacked_twin op 2022-04-28 13:41:47 +05:30
Utils importer: add initial support for loading BFloat16 tensors (#761) 2022-04-29 09:01:49 -07:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00