build image run on self host agent

Azure · Feb 4, 2024 · 8d7f58f · 8d7f58f
1 parent e7d63a9
commit 8d7f58f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/build-image.yaml b/.github/workflows/build-image.yaml
@@ -14,7 +14,7 @@ on:
 jobs:
   docker:
     name: Docker build ${{ matrix.name }}
-    runs-on: ubuntu-20.04
+    runs-on: [self-hosted, linux, x64, gpu]
     permissions:
       contents: read
       packages: write

diff --git a/dockerfile/torch1.14-cuda11.8.dockerfile b/dockerfile/torch1.14-cuda11.8.dockerfile
@@ -48,7 +48,7 @@ RUN cd third_party/msccl && \
     -gencode=arch=compute_90,code=sm_90" && \
     make install
 # cache TE build to save time in CI
-ENV MAX_JOBS=1
+# ENV MAX_JOBS=1
 RUN python3 -m pip install --upgrade pip && \
     python3 -m pip install git+https://github.com/NVIDIA/TransformerEngine.git@stable