torch-mlir/lib/Dialect/Torch/IR
Ashay Rane 88316b3b4e
torch: fold prim.dtype(bf16) to integer constant 15 (#1012)
A prior patch (63538de2) that added support for bfloat16 type did not
add the canonicalization pattern to fold `torch.prim.dtype` operations
on bfloat16 tensors into the integer constant 15.  This patch fixes the
problem.
2022-07-06 18:21:43 -07:00
..
CMakeLists.txt Reduce compilation time for TorchOps.cpp.inc 2022-03-21 14:42:26 -07:00
TorchDialect.cpp llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
TorchOps.cpp torch: fold prim.dtype(bf16) to integer constant 15 (#1012) 2022-07-06 18:21:43 -07:00
TorchOpsODSGenerated.cpp Reduce compilation time for TorchOps.cpp.inc 2022-03-21 14:42:26 -07:00
TorchTypes.cpp Add support for ScalarType::QUInt8 2022-06-29 15:33:28 -07:00
UtilsForODSGenerated.cpp llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
UtilsForODSGenerated.h Reduce compilation time for TorchOps.cpp.inc 2022-03-21 14:42:26 -07:00