Update opset version to 21 when importing onnx models (#329)

nod-ai · Aug 20, 2024 · 0fff0b9 · 0fff0b9
1 parent a555bf1
commit 0fff0b9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/e2eshark/run.py b/e2eshark/run.py
@@ -120,6 +120,7 @@ def runOnnxToTorchMLIRGeneration(
         logfilename = curphase + ".log"
         scriptcommand = (
             f"{torch_mlir_pythonpath} python -m torch_mlir.tools.import_onnx "
+            + "--opset-version=21 "
             + onnxfilename
             + " -o "
             + torchonnxfilename