test

worldcoin · Jul 19, 2024 · 5c16e38 · 5c16e38
1 parent 74141cf
commit 5c16e38
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/.github/workflows/test-gpu.yaml b/.github/workflows/test-gpu.yaml
@@ -10,6 +10,8 @@ concurrency:
 jobs:
   e2e:
     runs-on: gpu
+    container:
+      image:
     steps:
       - name: Checkout code
         uses: actions/checkout@v4
@@ -20,9 +22,6 @@ jobs:
       - name: Install OpenSSL && pkg-config
         run: sudo apt-get update && sudo apt-get install -y pkg-config libssl-dev
 
-      - name: Find libs
-        run: find /usr -name libnvrtc.so && find /usr -name libcuda.so
-
       - name: Install CUDA and NCCL dependencies
         if: steps.cache-cuda-nccl.outputs.cache-hit != 'true'
         env:
@@ -51,6 +50,9 @@ jobs:
           restore-keys: |
             rust-build-${{ runner.os }}-
 
+      - name: Find libs
+        run: find /usr -name libnvrtc.so && find /usr -name libcuda.so
+
       - name: Install Rust nightly
         uses: dtolnay/rust-toolchain@master
         with:
@@ -60,5 +62,5 @@ jobs:
         run: cargo test --release e2e
         shell: bash
         env:
-          NCCL_P2P_DIRECT_DISABLE: 1
+          NCCL_P2P_LEVEL: LOC
           NCCL_NET: Socket