torch-mlir icon indicating copy to clipboard operation
torch-mlir copied to clipboard

[ONNX] Dynamic size support for the onnx.GreaterOrEqual operator.

Open Peefy opened this issue 3 months ago • 1 comments

Hello, community. I have an ONNX model used the onnx.GreaterOrEqual operator. It encountered an error while converting to Torch.

onnx.mlir:1045:13: error: failed to legalize operation 'torch.operator' that was explicitly marked illegal
    %1041 = torch.operator "onnx.GreaterOrEqual"(%1040, %292) : (!torch.vtensor<[?],si64>, !torch.vtensor<[],si64>) -> !torch.vtensor<[?],i1> 
            ^
cvr.onnx.mlir:1045:13: note: see current operation: %7976 = "torch.operator"(%7975, %585) <{name = "onnx.GreaterOrEqual"}> : (!torch.vtensor<[?],si64>, !torch.vtensor<[],si64>) -> !torch.vtensor<[?],i1>

I noticed it failed here. https://github.com/llvm/torch-mlir/blob/main/lib/Conversion/TorchOnnxToTorch/DefaultDomainGtoP.cpp#L550

image

Is there a plan in the community to support this conversion, or can I participate in the community to implement it? Thank you!

Peefy avatar Apr 10 '24 08:04 Peefy