diff --git a/install.py b/install.py index 1686832..fd59f58 100644 --- 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: