[BYOC-DNNL] covered case for sum node without attr

apache · Aug 2, 2022 · 6428d35 · 6428d35
1 parent f9f2de7
commit 6428d35
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/tvm/relay/op/contrib/dnnl.py b/python/tvm/relay/op/contrib/dnnl.py
@@ -244,6 +244,8 @@ def predicate(expr):
         for e, op_name in zip([expr, expr.args[0]], ["sum", "bias_add"]):
             args = get_args(e)
             attrs = get_attrs(e.args[0])
+            if attrs is None:
+                return False
             if not checker(attrs, args, op_name):
                 return False
         return True