From ea2083bd12d5e646f711d9b6b76acb9ae3b40a20 Mon Sep 17 00:00:00 2001
From: wojciechsromek <wojciech.sromek@toolsforhumanity.com>
Date: Fri, 19 Jul 2024 12:47:06 +0200
Subject: [PATCH] test

---
 .github/workflows/test-gpu.yaml | 53 ++++++++++++++-------------------
 1 file changed, 22 insertions(+), 31 deletions(-)

diff --git a/.github/workflows/test-gpu.yaml b/.github/workflows/test-gpu.yaml
index 10bfa8aa5..8c495d022 100644
--- a/.github/workflows/test-gpu.yaml
+++ b/.github/workflows/test-gpu.yaml
@@ -14,35 +14,26 @@ jobs:
       - name: Checkout code
         uses: actions/checkout@v4
 
-      - run: find /usr -name "libcuda*"
-      - run: nvidia-smi
-      - run: sleep 300
+      - name: Cache Rust build
+        uses: actions/cache@v3
+        id: cache-rust
+        with:
+          path: |
+            ~/.cargo/registry
+            ~/.cargo/git
+            target
+          key: rust-build-${{ runner.os }}-${{ hashFiles('**/Cargo.lock') }}
+          restore-keys: |
+            rust-build-${{ runner.os }}-
 
-#      - name: Cache Rust build
-#        uses: actions/cache@v3
-#        id: cache-rust
-#        with:
-#          path: |
-#            ~/.cargo/registry
-#            ~/.cargo/git
-#            target
-#          key: rust-build-${{ runner.os }}-${{ hashFiles('**/Cargo.lock') }}
-#          restore-keys: |
-#            rust-build-${{ runner.os }}-
-#
-#      - name: Install Rust nightly
-#        uses: dtolnay/rust-toolchain@master
-#        with:
-#          toolchain: nightly
-#
-#      - name: E2E Tests
-#        run: |
-#          ls -al /usr/lib
-#          export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64:/usr/lib/x86_64-unknown-linux-gnu:$LD_LIBRARY_PATH
-#          export CUDA_HOME=/usr/local/cuda-12.2
-#          /usr/local/cuda-12.2/bin/nvcc --version
-#          cargo test --release e2e
-#        shell: bash
-#        env:
-#          NCCL_P2P_DIRECT_DISABLE: 1
-#          NCCL_NET: Socket
+      - name: Install Rust nightly
+        uses: dtolnay/rust-toolchain@master
+        with:
+          toolchain: nightly
+
+      - name: E2E Tests
+        run: cargo test --release e2e
+        shell: bash
+        env:
+          NCCL_P2P_DIRECT_DISABLE: 1
+          NCCL_NET: Socket