iree icon indicating copy to clipboard operation
iree copied to clipboard

failed to legalize operation 'torch.aten.arange.start_step'

Open alexsifivetw opened this issue 1 year ago • 3 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

71a9945

Additional context

No response

alexsifivetw avatar May 16 '24 03:05 alexsifivetw

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 🤔).

ScottTodd avatar May 16 '24 03:05 ScottTodd

I ran into the same issue with iree-compile from pip, but it works with a build of main.

caojoshua avatar Jun 08 '24 07:06 caojoshua

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...

ScottTodd avatar Jun 10 '24 15:06 ScottTodd