From 0cc512bfb20b4c81dc58ea3a8db5d5fd78deb77e Mon Sep 17 00:00:00 2001 From: Xu Zhao Date: Thu, 19 Dec 2024 17:34:47 -0500 Subject: [PATCH] Set memory limit in docker build --- .github/workflows/docker.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/docker.yaml b/.github/workflows/docker.yaml index a6c930a..fa7b1f4 100644 --- a/.github/workflows/docker.yaml +++ b/.github/workflows/docker.yaml @@ -42,7 +42,7 @@ jobs: # branch name is github.head_ref when triggered by pull_request # and it is github.ref_name when triggered by workflow_dispatch branch_name=${{ github.head_ref || github.ref_name }} - docker build . --memory 100g --build-arg TRITONBENCH_BRANCH="${branch_name}" --build-arg FORCE_DATE="${NIGHTLY_DATE}" \ + docker build . --memory 80g --shm-size 4g --build-arg TRITONBENCH_BRANCH="${branch_name}" --build-arg FORCE_DATE="${NIGHTLY_DATE}" \ -f tritonbench-nightly.dockerfile -t ghcr.io/pytorch-labs/tritonbench:latest # Extract pytorch version from the docker PYTORCH_VERSION=$(docker run -e SETUP_SCRIPT="${SETUP_SCRIPT}" ghcr.io/pytorch-labs/tritonbench:latest bash -c '. "${SETUP_SCRIPT}"; python -c "import torch; print(torch.__version__)"')