diff --git a/install.py b/install.py index 2fa7a9e..9658a14 100644 --- 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