iree
iree copied to clipboard
failed to legalize operation 'torch.aten.arange.start_step'
What happened?
When trying to compile the mlir file, I encountered the error below:
error: failed to legalize operation 'torch.aten.arange.start_step' that was explicitly marked illegal
%8 = torch.aten.triu %7, %int1_10 : !torch.vtensor<[8,9],f32>, !torch.int -> !torch.vtensor<[8,9],f32>
^
Steps to reproduce your issue
No response
What component(s) does this issue relate to?
No response
Version information
71a9945
Additional context
No response
Looks like triu is not implemented in https://github.com/llvm/torch-mlir . We also have test failures for onnx.Trilu that would use a similar lowering path (those look like the same math 🤔).
I ran into the same issue with iree-compile from pip, but it works with a build of main.
Ah, some triu tests did start passing at https://github.com/iree-org/iree/commit/440c87019240e225bafce595be5cf61ce344e0c8. Also yes, we should push a new version to pip soon...