lint

apache · Dec 22, 2021 · 56e0e95 · 56e0e95
1 parent d47c100
commit 56e0e95
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/tvm/relay/op/contrib/cutlass.py b/python/tvm/relay/op/contrib/cutlass.py
@@ -73,11 +73,11 @@ def make_conv2d_pattern(with_bias=False, with_act=None):
     if with_act is not None:
         if with_act == "relu":
             return is_op("nn.relu")(conv2d_out)
-        elif with_act == "sigmoid":
+        if with_act == "sigmoid":
             return is_op("sigmoid")(conv2d_out)
-        elif with_act == "silu":
+        if with_act == "silu":
             return is_op("multiply")(conv2d_out, is_op("sigmoid")(conv2d_out))
-        elif with_act == "hardswish":
+        if with_act == "hardswish":
             rhs = is_op("divide")(
                 is_op("clip")(is_op("add")(conv2d_out, is_constant())), is_constant()
             )