Disable colfax cutlass

pytorch-labs · Dec 19, 2024 · bdea110 · bdea110
1 parent 9a9a342
commit bdea110
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/install.py b/install.py
@@ -131,7 +131,7 @@ def setup_hip(args: argparse.Namespace):
     if args.fa3 or args.all:
         logger.info("[tritonbench] installing fa3...")
         install_fa3()
-    if args.colfax or args.all:
+    if args.colfax:
         logger.info("[tritonbench] installing colfax cutlass-kernels...")
         from tools.cutlass_kernels.install import install_colfax_cutlass