torch-mlir/lib/Dialect/Torch
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
..
IR torch: fold prim.dtype(bf16) to integer constant 15 (#1012) 2022-07-06 18:21:43 -07:00
Transforms Fix compilation warning Wsign-compare (#1003) 2022-07-06 09:06:10 -07:00
Utils Fix type promotion code for scalar only operations 2022-05-07 10:37:21 -04:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00