Compile fa2 but not xformers

pytorch-labs · Dec 20, 2024 · ed90ab1 · ed90ab1
1 parent 1e03e04
commit ed90ab1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/install.py b/install.py
@@ -121,7 +121,7 @@ def setup_hip(args: argparse.Namespace):
     if args.fbgemm or args.all:
         logger.info("[tritonbench] installing FBGEMM...")
         install_fbgemm()
-    if args.fa2:
+    if args.fa2 or args.all:
         logger.info("[tritonbench] installing fa2 from source...")
         install_fa2(compile=True)
     if args.colfax:
@@ -140,7 +140,7 @@ def setup_hip(args: argparse.Namespace):
     if args.liger or args.all:
         logger.info("[tritonbench] installing liger-kernels...")
         install_liger()
-    if args.xformers or args.all:
+    if args.xformers:
         logger.info("[tritonbench] installing xformers...")
         from tools.xformers.install import install_xformers