Skip fa2

pytorch-labs · Dec 20, 2024 · 1e03e04 · 1e03e04
1 parent b413944
commit 1e03e04
Showing 1 changed file with 1 addition and 1 deletion.
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 or args.all:
+    if args.fa2:
         logger.info("[tritonbench] installing fa2 from source...")
         install_fa2(compile=True)
     if args.colfax: