Another fix

pytorch · Jun 12, 2024 · 7e6c79d · 7e6c79d
1 parent aeaaa0d
commit 7e6c79d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/docker/torchbench-nightly.dockerfile b/docker/torchbench-nightly.dockerfile
@@ -9,10 +9,8 @@ ARG TORCHBENCH_BRANCH=${TORCHBENCH_BRANCH:-main}
 ARG FORCE_DATE=${FORCE_DATE}
 
 # Checkout Torchbench and submodules
-RUN git clone -b "${TORCHBENCH_BRANCH}" --single-branch \
+RUN git clone --recurse-submodules -b "${TORCHBENCH_BRANCH}" --single-branch \
     https://github.com/pytorch/benchmark /workspace/benchmark
-RUN cd /workspace/benchmark \
-    git submodule update --init --recursive
 
 # Setup conda env and CUDA
 RUN cd /workspace/benchmark && \

diff --git a/userbenchmark/triton/install.py b/userbenchmark/triton/install.py
@@ -8,6 +8,8 @@
 FBGEMM_PATH = REPO_PATH.joinpath("submodules", "FBGEMM", "fbgemm_gpu")
 
 def install_fbgemm():
+    cmd = ["pip", "install", "-r", "requirements.txt"]
+    subprocess.check_call(cmd, cwd=str(FBGEMM_PATH.resolve()))
     cmd = [sys.executable, "setup.py", "bdist_wheel", "--package_variant=genai"]
     subprocess.check_call(cmd, cwd=str(FBGEMM_PATH.resolve()))