llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Dialect/Torch
History
Tanyo Kwok 290d7755fb
importer: add initial support for loading Float16 tensors (#1169) 
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.
 		2022-08-08 12:37:31 +08:00
..
IR [MLIR][TORCH] Add support for multiple indexing tensors for aten.index.Tensor (#1097) 2022-07-28 19:00:02 -04:00
Transforms [MHLO] fix tensor mode aten.div op pattern (#1160) 2022-08-06 23:38:06 +08:00
Utils importer: add initial support for loading Float16 tensors (#1169) 2022-08-08 12:37:31 +08:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00