mirror of https://github.com/llvm/torch-mlir
55ff110dc2
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
|
||
---|---|---|
.. | ||
IR | ||
Transforms | ||
Utils | ||
CMakeLists.txt |