fixed device issue

Signed-off-by: Elias Joseph <[email protected]>
nod-ai · Feb 15, 2025 · 06175a7 · 06175a7
1 parent 69da7c9
commit 06175a7
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 7 deletions.
diff --git a/.github/workflows/ci-sglang-integration-tests.yml b/.github/workflows/ci-sglang-integration-tests.yml
@@ -69,12 +69,6 @@ jobs:
           pip install sentence_transformers
 
           pip freeze
-
-      - name: run rocm tests
-        run: |
-          rocminfo
-          rocm-smi
-
       - name: Run Integration Tests
         run: |
           source ${VENV_DIR}/bin/activate

diff --git a/app_tests/benchmark_tests/llm/sglang_benchmarks/shortfin_benchmark_test.py b/app_tests/benchmark_tests/llm/sglang_benchmarks/shortfin_benchmark_test.py
@@ -60,7 +60,7 @@ def test_shortfin_benchmark(
     request,
 ):
     # TODO: Remove when multi-device is fixed
-    os.environ["ROCR_VISIBLE_DEVICES"] = "1"
+    os.environ["ROCR_VISIBLE_DEVICES"] = "0"
 
     process, port = server