From 6428d35f1b77bf3a114b92812e0ce8696a9b899d Mon Sep 17 00:00:00 2001 From: "Michael J. Klaiber" Date: Tue, 2 Aug 2022 10:20:15 +0000 Subject: [PATCH] [BYOC-DNNL] covered case for sum node without attr --- python/tvm/relay/op/contrib/dnnl.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/tvm/relay/op/contrib/dnnl.py b/python/tvm/relay/op/contrib/dnnl.py index 46c20e947f01..e7ed49d4ccad 100644 --- 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