61
61
parameters :
62
62
Dockerfile : tools/ci_build/github/linux/docker/rocm-ci-pipeline-env.Dockerfile
63
63
Context : tools/ci_build/github/linux/docker
64
- DockerBuildArgs : " --build-arg ROCM_VERSION=$(RocmVersion)"
65
- Repository : onnxruntimerocm-cibuild-rocm$(RocmVersion)
64
+ DockerBuildArgs : " --build-arg ROCM_VERSION=$(RocmVersion) --build-arg BUILD_UID=1004 "
65
+ Repository : onnxruntimerocm-cibuild-rocm$(RocmVersion)new
66
66
67
67
- task : Cache@2
68
68
inputs :
90
90
--volume $(CCACHE_DIR):/cache \
91
91
-e CCACHE_DIR=/cache \
92
92
--workdir /onnxruntime_src \
93
- onnxruntimerocm-cibuild-rocm$(RocmVersion) \
93
+ onnxruntimerocm-cibuild-rocm$(RocmVersion)new \
94
94
/bin/bash -c "
95
95
set -ex; \
96
96
env; \
@@ -161,8 +161,8 @@ jobs:
161
161
parameters :
162
162
Dockerfile : tools/ci_build/github/linux/docker/rocm-ci-pipeline-env.Dockerfile
163
163
Context : tools/ci_build/github/linux/docker
164
- DockerBuildArgs : " --build-arg ROCM_VERSION=$(RocmVersion)"
165
- Repository : onnxruntimerocm-cibuild-rocm$(RocmVersion)
164
+ DockerBuildArgs : " --build-arg ROCM_VERSION=$(RocmVersion) --build-arg BUILD_UID=1004 "
165
+ Repository : onnxruntimerocm-cibuild-rocm$(RocmVersion)new
166
166
167
167
- task : CmdLine@2
168
168
inputs :
@@ -179,7 +179,7 @@ jobs:
179
179
--volume $(Build.BinariesDirectory):/build \
180
180
--volume /data/models:/build/models:ro \
181
181
--workdir /build/Release \
182
- onnxruntimerocm-cibuild-rocm$(RocmVersion) \
182
+ onnxruntimerocm-cibuild-rocm$(RocmVersion)new \
183
183
/bin/bash -c "
184
184
set -ex; \
185
185
xargs -a /build/Release/perms.txt chmod a+x; \
@@ -225,10 +225,10 @@ jobs:
225
225
-e KERNEL_EXPLORER_BATCHED_GEMM_MAX_BATCH_SIZE=8 \
226
226
-e KERNEL_EXPLORER_TEST_USE_CUPY=0 \
227
227
-e CUPY_CACHE_DIR=/build/Release \
228
- onnxruntimerocm-cibuild-rocm$(RocmVersion) \
228
+ onnxruntimerocm-cibuild-rocm$(RocmVersion)new \
229
229
/bin/bash -c "
230
230
set -ex; \
231
- python --version; \
231
+ python --version; id ; ls -lha /home ; \
232
232
ls /opt/miniconda/envs/rocm-ci/lib/; \
233
233
pytest /onnxruntime_src/onnxruntime/python/tools/kernel_explorer/ -n 4 --reruns 1 --durations=100"
234
234
workingDirectory : $(Build.SourcesDirectory)
0 commit comments