fix

masahi · masahi · commit 6de3d6429b90 · 2022-01-26T09:54:11.000+09:00
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:

Original file line number	Diff line number	Diff line change
`@@ -217,7 +217,7 @@ def select_op(`
`217`	`217`	`profile_all_alignments,`
`218`	`218`	`)`
`219`	`219`
`220`		`- if find_first_valid:`
	`220`	`+ if not find_first_valid:`
`221`	`221`	`self.engine.compile_all(ops, use_multiprocessing)`
`222`	`222`
`223`	`223`	`args = (`
Original file line number	Diff line number	Diff line change
`@@ -222,7 +222,7 @@ def select_op(`
`222`	`222`	`profile_all_alignments=profile_all_alignments,`
`223`	`223`	`)`
`224`	`224`
`225`		`- if find_first_valid:`
	`225`	`+ if not find_first_valid:`
`226`	`226`	`self.engine.compile_all(ops, use_multiprocessing)`
`227`	`227`
`228`	`228`	`for op in ops:`