[Bugfix][Cutlass] Remove a typo in cutlass build (#16789)

Introduced in #16745, should be the string `"Composite"`, not the bytes `b"Composite"`.
apache · Mar 26, 2024 · bcfbcab · bcfbcab
1 parent bf2d43e
commit bcfbcab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 b"Composite" not in f.attrs:
+        if "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)