[Relay] Check if the attribute "name" exists before accessing it

apache · Apr 4, 2023 · c341e1a · c341e1a
1 parent f5db8b7
commit c341e1a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/qnn/op/_qnn.py b/python/tvm/relay/qnn/op/_qnn.py
@@ -101,7 +101,7 @@ def legalize_clip(attrs, inputs, tinfos):
     implementations (like Cortex-M) need it to be done earlier in legalization.
     """
 
-    if hasattr(inputs[0], "op") and inputs[0].op.name == "qnn.requantize":
+    if hasattr(inputs[0], "op") and hasattr(inputs[0].op, "name") and inputs[0].op.name == "qnn.requantize":
         dtype_info = np.iinfo(tinfos[0].dtype)
         if dtype_info.min == attrs.a_min and dtype_info.max == attrs.a_max:
             return inputs[0]