diff --git a/python/tvm/contrib/cutlass/build.py b/python/tvm/contrib/cutlass/build.py index 1c0a30c62d91..80169f51640e 100644 --- a/python/tvm/contrib/cutlass/build.py +++ b/python/tvm/contrib/cutlass/build.py @@ -977,7 +977,7 @@ def handle_norm(self, f, op_type): return f.with_attrs(attrs) def visit_function_(self, f): - if "Composite" not in f.attrs: + if f.attrs is None or "Composite" not in f.attrs: body = super().visit_expr(f.body) return relax.Function(f.params, body, f.ret_struct_info, f.is_pure, f.attrs, f.span)