fix

apache · Jan 26, 2022 · 6de3d64 · 6de3d64
1 parent d15a995
commit 6de3d64
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/tvm/contrib/cutlass/gen_conv2d.py b/python/tvm/contrib/cutlass/gen_conv2d.py
@@ -217,7 +217,7 @@ def select_op(
             profile_all_alignments,
         )
 
-        if find_first_valid:
+        if not find_first_valid:
             self.engine.compile_all(ops, use_multiprocessing)
 
         args = (

diff --git a/python/tvm/contrib/cutlass/gen_gemm.py b/python/tvm/contrib/cutlass/gen_gemm.py
@@ -222,7 +222,7 @@ def select_op(
             profile_all_alignments=profile_all_alignments,
         )
 
-        if find_first_valid:
+        if not find_first_valid:
             self.engine.compile_all(ops, use_multiprocessing)
 
         for op in ops: