[ROCm] Update ROCm and MigraphX CI to ROCm5.4 (#14011)

Update ROCm and MigraphX CI to ROCm5.4
Run ortmodule_test with ROCm5.4 and all
passed(https://dev.azure.com/onnxruntime/onnxruntime/_build/results?buildId=824742&view=logs&j=8292f886-7946-5da9-7977-04484c342eda&t=5de68eaa-cbdc-5be5-13d0-bb946f4ddb2d).

Co-authored-by: peixuanzuo <peixuanzuo@linmif39a000004.zvflicr54joexhdgnhvmxrxygg.phxx.internal.cloudapp.net>
This commit is contained in:
PeixuanZuo 2022-12-22 10:01:05 +08:00 committed by GitHub
parent df8ff34f25
commit ab2dd8dfaf
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 7 additions and 7 deletions

View file

@ -8,14 +8,14 @@ jobs:
pool: 'AMD-GPU'
timeoutInMinutes: 180
# gid of video and render group on gcr-openpai-35 and -36
# gid of video and render group on gcramdrr1-mi100-085 and -86
variables:
- name: video
value: 44
- name: render
value: 109
- name: RocmVersion
value: 5.2.3
value: 5.4
steps:
- checkout: self
@ -60,7 +60,7 @@ jobs:
--parallel 32 \
--build_wheel \
--skip_submodule_sync \
--skip_tests --cmake_path /usr/bin/cmake --ctest_path /usr/bin/ctest
--skip_tests --cmake_path /usr/bin/cmake --ctest_path /usr/bin/ctest
workingDirectory: $(Build.SourcesDirectory)
displayName: 'Build onnxruntime'

View file

@ -8,7 +8,7 @@ jobs:
pool: 'AMD-GPU'
timeoutInMinutes: 150
# gid of video and render group on gcr-openpai-35 and -36
# gid of video and render group on gcramdrr1-mi100-085 and -86
variables:
- name: video
value: 44
@ -17,7 +17,7 @@ jobs:
- name: onnxruntimeBuildSucceeded
value: false
- name: RocmVersion
value: 5.2.3
value: 5.4
# generated from tools/ci_build/github/pai/rocm-ci-pipeline-env.Dockerfile
container:

View file

@ -1,4 +1,4 @@
FROM rocm/pytorch:rocm5.2.3_ubuntu20.04_py3.7_pytorch_1.12.1
FROM rocm/pytorch:rocm5.4_ubuntu20.04_py3.7_pytorch_1.12.1
ENV DEBIAN_FRONTEND noninteractive
ENV MIGRAPHX_DISABLE_FAST_GELU=1

View file

@ -1,4 +1,4 @@
FROM rocm/pytorch:rocm5.2.3_ubuntu20.04_py3.7_pytorch_1.12.1
FROM rocm/pytorch:rocm5.4_ubuntu20.04_py3.7_pytorch_1.12.1
WORKDIR /stage