From 7a253eb872cc21d3577516c964e64bd531d41d38 Mon Sep 17 00:00:00 2001
From: elenaslavutina <elena.slavutina2013@gmail.com>
Date: Mon, 28 Jun 2021 12:27:28 +0300
Subject: [PATCH 1/2] add warning in case of opset mismatch

---
 python/tvm/relay/frontend/onnx.py | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py
index 5d07102f2c3f..14531d7a0505 100644
--- a/python/tvm/relay/frontend/onnx.py
+++ b/python/tvm/relay/frontend/onnx.py
@@ -3617,11 +3617,21 @@ def from_onnx(model, shape=None, dtype="float32", opset=None, freeze_params=Fals
         pass
     g = GraphProto(shape, dtype, freeze_params)
     graph = model.graph
+
+    try:
+        opset_in_model = model.opset_import[0].version if model.opset_import else 1
+    except AttributeError:
+        opset_in_model = 1
+
     if opset is None:
-        try:
-            opset = model.opset_import[0].version if model.opset_import else 1
-        except AttributeError:
-            opset = 1
+        opset = opset_in_model
+    elif opset < opset_in_model:
+        warnings.warn(
+                      ""
+                      f"You are overwritting original opset ver = {opset_in_model} by lower ver = {opset}. "
+                      f"That might cause model conversion errors."
+        )
+
     # Use the graph proto as a scope so that ops can access other nodes if needed.
     with g:
         mod, params = g.from_onnx(graph, opset)

From ceac8f1f2defa8514c65f7bc8c0811adb3b83ad5 Mon Sep 17 00:00:00 2001
From: islavutin <iliya.slavutin@gmail.com>
Date: Mon, 28 Jun 2021 16:02:34 +0300
Subject: [PATCH 2/2] fix CI

---
 python/tvm/relay/frontend/onnx.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py
index 14531d7a0505..f4a71a90e17f 100644
--- a/python/tvm/relay/frontend/onnx.py
+++ b/python/tvm/relay/frontend/onnx.py
@@ -3627,9 +3627,9 @@ def from_onnx(model, shape=None, dtype="float32", opset=None, freeze_params=Fals
         opset = opset_in_model
     elif opset < opset_in_model:
         warnings.warn(
-                      ""
-                      f"You are overwritting original opset ver = {opset_in_model} by lower ver = {opset}. "
-                      f"That might cause model conversion errors."
+            ""
+            f"You are overwritting original opset ver = {opset_in_model} by lower ver = {opset}. "
+            f"That might cause model conversion errors."
         )
 
     # Use the graph proto as a scope so that ops can access other nodes if needed.