change is_constant pattern to wildcard in gelu pattern

masahi · Nov 3, 2021 · cfacfa2 · cfacfa2
1 parent 84da943
commit cfacfa2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/op/contrib/cutlass.py b/python/tvm/relay/op/contrib/cutlass.py
@@ -20,7 +20,7 @@
 
 
 def make_gelu_pattern(bias_out, out_dtype="float16"):
-    mul = is_op("multiply")(bias_out, is_constant())
+    mul = is_op("multiply")(bias_out, wildcard())
     if out_dtype == "float16":
         erf = is_op("cast")(is_op("erf")(is_op("cast")(mul)))
     else: