From 0fff0b9b2df7611c5d4a122a12b89e9e65f2ca1f Mon Sep 17 00:00:00 2001
From: jinchen <49575973+jinchen62@users.noreply.github.com>
Date: Tue, 20 Aug 2024 14:13:31 -0700
Subject: [PATCH] Update opset version to 21 when importing onnx models (#329)

---
 e2eshark/run.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/e2eshark/run.py b/e2eshark/run.py
index 8742282f..c0729944 100644
--- 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