@@ -108,8 +108,6 @@ jobs:
108108 cat /tmp/env_file.txt
109109
110110 docker run -dt --device=/dev/kfd $DEVICE_FLAG \
111- -v "${GITHUB_WORKSPACE:-$PWD}":/workspace \
112- -w /workspace \
113111 --ipc=host --group-add video \
114112 --shm-size=16G \
115113 --privileged \
@@ -121,19 +119,19 @@ jobs:
121119 -e ATOM_DISABLE_MMAP=true \
122120 -v "${{ github.workspace }}:/workspace" \
123121 -w /workspace \
124- --name atom_test \
122+ --name atom_aiter_test \
125123 rocm/aiter-ci:atom-test
126124 env :
127125 GITHUB_WORKSPACE : ${{ github.workspace }}
128126
129127 - name : Check shm size
130128 run : |
131129 df -h /dev/shm
132- docker exec atom_test df -h /dev/shm
130+ docker exec atom_aiter_test df -h /dev/shm
133131
134132 - name : Check version
135133 run : |
136- docker exec atom_test bash -lc "pip show amd-aiter atom && pip list"
134+ docker exec atom_aiter_test bash -lc "pip show amd-aiter atom && pip list"
137135
138136 - name : Run ATOM simple inference
139137 timeout-minutes : 30
@@ -146,7 +144,7 @@ jobs:
146144 model_path="${{ matrix.model_path }}"
147145 echo "Model path : $model_path"
148146
149- docker exec atom_test bash -lc "
147+ docker exec atom_aiter_test bash -lc "
150148 set -euo pipefail
151149 python3 -m atom.examples.simple_inference \
152150 --model \"$model_path\" \
@@ -166,12 +164,12 @@ jobs:
166164 echo ""
167165 echo "========== Launching ATOM server =========="
168166 model_path="${{ matrix.model_path }}"
169- docker exec atom_test bash -lc "
167+ docker exec atom_aiter_test bash -lc "
170168 .github/scripts/atom_test.sh launch $model_path ${{ matrix.extraArgs }}
171169 "
172170 echo ""
173171 echo "========== Running accuracy test =========="
174- docker exec atom_test bash -lc "
172+ docker exec atom_aiter_test bash -lc "
175173 .github/scripts/atom_test.sh accuracy $model_path
176174 "
177175
@@ -185,5 +183,5 @@ jobs:
185183 - name : Clean Up
186184 if : always()
187185 run : |
188- docker stop atom_test || true
189- docker rm atom_test || true
186+ docker stop atom_aiter_test || true
187+ docker rm atom_aiter_test || true
0 commit comments