failed to legalize operation 'torch.aten.arange.start_step'
alexsifivetw opened this issue · comments
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
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 440c870. Also yes, we should push a new version to pip soon...