torch-mlir icon indicating copy to clipboard operation
torch-mlir copied to clipboard
verified publisher icon llvm

Wrap indices to be within max range for as_strided + index_select

Open sahas3 opened this issue 1 year ago • 1 comments

Found these issues when trying to lower the ExportedProgram generated from decomposing nn.GRU op in torch.

  • Added a new e2e test representing the root cause which involves a special configuration of as_strided to reproduce the issue. It requires the other ops in the repro to hit the issue.
  • For --config=onnx, index_select op was generated instead of as_strided which had the same issue as well.

notify : @justin-ngo-arm as you were the original author for the Tosa conversions of these ops.

sahas3 avatar Dec 03 '24 15:12 sahas3

@sahas3 Please rebase your PR, so that the CI can be run.

Rebased PR and CI is clean. Thanks!

sahas3 avatar Dec 23 '24 19:12 sahas3