nod-ai / SHARK-ModelDev

Unified compiler/runtime for interfacing with PyTorch Dynamo.
Apache License 2.0
95 stars 48 forks source link

Failed to legalize operation `onnx.NonZero`. #820

Open vinayakdsci opened 2 months ago

vinayakdsci commented 2 months ago

The ONNX operation onnx.NonZero fails with failed to legalize operation 'onnx.NonZero' during lowering, as reported in https://github.com/nod-ai/SHARK-Turbine/issues/812.

More details will be added later.

renxida commented 1 month ago

working on it here: https://github.com/llvm/torch-mlir/issues/3293