[ROCm] Fix CI Pipeline (#16409)

1. add `set -ex` before commands.
2. update ccache.
This commit is contained in:
PeixuanZuo 2023-06-19 15:22:13 +08:00 committed by GitHub
parent 8b9eab093b
commit 1418d8728c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -86,8 +86,9 @@ jobs:
-e CCACHE_DIR=/cache \
--workdir /onnxruntime_src \
onnxruntimetrainingrocm-cibuild-rocm$(RocmVersion) \
/bin/bash -c "
ccache -s && ccache -z; \
/bin/bash -c "
set -ex; \
ccache -s; \
python tools/ci_build/build.py \
--config $(BuildConfig) \
--enable_training \
@ -153,6 +154,7 @@ jobs:
--volume $(Build.BinariesDirectory):/build \
onnxruntimetrainingrocm-cibuild-rocm$(RocmVersion) \
/bin/bash -c "
set -ex; \
export KERNEL_EXPLORER_BUILD_DIR=/build/$(BuildConfig); \
pytest /onnxruntime_src/onnxruntime/python/tools/kernel_explorer/ -n 16 --reruns 1"
workingDirectory: $(Build.SourcesDirectory)
@ -175,6 +177,7 @@ jobs:
--workdir /build/$(BuildConfig) \
onnxruntimetrainingrocm-cibuild-rocm$(RocmVersion) \
/bin/bash -c "
set -ex; \
export PYTHONPATH=/build/$(BuildConfig); \
python -m onnxruntime.training.ortmodule.torch_cpp_extensions.install; \
python \
@ -238,6 +241,7 @@ jobs:
--workdir /build/$(BuildConfig) \
onnxruntimetrainingrocm-cibuild-rocm$(RocmVersion) \
/bin/bash -c "
set -ex; \
unset PYTHONPATH; \
pip install /build/$(BuildConfig)/dist/$whlfilename; \
python -m onnxruntime.training.ortmodule.torch_cpp_extensions.install; \