torch-mlir/lib/Dialect/Torch
zjgarvey 55ff110dc2
[MLIR][TORCH] Only unroll prim loop-like ops within a `torch.shape.calculate` region (#3812)
Reports a match failure for the pattern `FullyUnrollPrimLoop` when the
loop op is not in a region defined by a `torch.shape.calculate` op.

This is needed to avoid unrolling prim loops generated by ONNX IR, since
we are applying shape refinement in the
`torch-onnx-to-torch-backend-pipeline` introduced in fa4794d .

See also the discussion in
<https://github.com/iree-org/iree/pull/18867#discussion_r1811101655>
2024-10-23 13:38:55 +05:30
..
IR Add More Scalarize Shapes Patterns (#3810) 2024-10-21 19:42:39 -05:00
Transforms [MLIR][TORCH] Only unroll prim loop-like ops within a `torch.shape.calculate` region (#3812) 2024-10-23 13:38:55 +05:30
Utils Add Op for `torch.aten.unfold` (#3772) 2024-10-08 21:10:43 +00:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00