From 0100a13a13e7234068d808575c4e42de07ff54ca Mon Sep 17 00:00:00 2001
From: CSY <csy@modelcloud.ai>
Date: Wed, 4 Dec 2024 09:32:19 +0800
Subject: [PATCH] use RUNNER to set ip

---
 .github/workflows/unit_tests.yml | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml
index f86288a..e63fc90 100644
--- a/.github/workflows/unit_tests.yml
+++ b/.github/workflows/unit_tests.yml
@@ -9,6 +9,10 @@ on:
   repository_dispatch:
   workflow_dispatch:
 
+env:
+  CUDA_DEVICE_ORDER: PCI_BUS_ID
+  RUNNER: 10.0.14.248
+
 jobs:
   build:
     runs-on: ${{ matrix.os }}
@@ -69,7 +73,7 @@ jobs:
   gpu:
     runs-on: self-hosted
     container:
-      image: 10.0.14.248:5000/modelcloud/gptqmodel:compiler_cuda124-torch2.5.1-python311
+      image: ${RUNNER}:5000/modelcloud/gptqmodel:compiler_cuda124-torch2.5.1-python311
     steps:
       - uses: actions/checkout@v4