From 9a9a342eee199e95c1bc4c340dd85a8dec65edca Mon Sep 17 00:00:00 2001
From: Xu Zhao <i@xuzhao.net>
Date: Thu, 19 Dec 2024 00:26:53 -0500
Subject: [PATCH] Test

---
 .ci/tritonbench/install.sh | 6 +-----
 install.py                 | 2 +-
 2 files changed, 2 insertions(+), 6 deletions(-)

diff --git a/.ci/tritonbench/install.sh b/.ci/tritonbench/install.sh
index a848238b..c30ab81c 100644
--- a/.ci/tritonbench/install.sh
+++ b/.ci/tritonbench/install.sh
@@ -10,9 +10,5 @@ fi
 tritonbench_dir=$(dirname "$(readlink -f "$0")")/../..
 cd ${tritonbench_dir}
 
-# probe memory available
-free -h
-
 # Install Tritonbench and all its customized packages
-# Test: only install fa3
-python install.py --fa3
+python install.py --all
diff --git a/install.py b/install.py
index ab6ded70..2fa7a9e4 100644
--- a/install.py
+++ b/install.py
@@ -69,7 +69,7 @@ def install_fa3():
     FA3_PATH = REPO_PATH.joinpath("submodules", "flash-attention", "hopper")
     env = os.environ.copy()
     # nvcc will now spawn cicc and will cost ~1G memory
-    env["MAX_JOBS"] = "4"
+    env["MAX_JOBS"] = "8"
     cmd = [sys.executable, "setup.py", "install"]
     subprocess.check_call(cmd, cwd=str(FA3_PATH.resolve()), env=env)