[ci] fixed booster test

hpcaitech · Jan 11, 2024 · 0d8af5c · 0d8af5c
1 parent 7e2c7e4
commit 0d8af5c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/.github/workflows/build_on_pr.yml b/.github/workflows/build_on_pr.yml
@@ -90,7 +90,7 @@ jobs:
     runs-on: [self-hosted, gpu]
     container:
       image: hpcaitech/pytorch-cuda:2.1.0-12.1.0
-      options: --gpus all --rm --shm-size=512m -v /data/scratch/llama-tiny:/data/scratch/llama-tiny
+      options: --gpus all --rm -v /dev/shm -v /data/scratch/llama-tiny:/data/scratch/llama-tiny
     timeout-minutes: 60
     defaults:
       run:
@@ -168,7 +168,6 @@ jobs:
           # --ignore tests/test_smoothquant \
           # --ignore tests/test_checkpoint_io \
         env:
-          NCCL_SHM_DISABLE: 1
           LD_LIBRARY_PATH: /github/home/.tensornvme/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64
           LLAMA_PATH: /data/scratch/llama-tiny