From 7e6c79df0efad75f3ae9c428202ad1b890f16e97 Mon Sep 17 00:00:00 2001 From: Xu Zhao Date: Wed, 12 Jun 2024 10:31:01 -0400 Subject: [PATCH] Another fix --- docker/torchbench-nightly.dockerfile | 4 +--- userbenchmark/triton/install.py | 2 ++ 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docker/torchbench-nightly.dockerfile b/docker/torchbench-nightly.dockerfile index a35a3dda20..f6f574af8e 100644 --- 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 index 360806c3d2..21c61dc802 100644 --- 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()))