saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Switch AMD CI pipeline to use environment image from onnxruntimecibuildenvironment (#9206)

Browse source 
* shift docker image reference for amd ci pipeline

* fix service endpoint

* reduce perf tolerance
This commit is contained in:
Suffian Khan 2021-09-28 13:06:16 -07:00 committed by GitHub
parent 1104e8d3e5
commit 6f580f07de
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
orttraining/tools/ci_test/compare_huggingface.py
View file

@ -33,7 +33,7 @@ for i in range(logged_steps):
success = all(loss_tail_matches)
# performance match
threshold = 0.97
threshold = 0.95
is_performant = json_actual['samples_per_second'] >= threshold*json_expect['samples_per_second']
success = success if is_performant else False
print('samples_per_second actual {:.3f} expected {:.3f} in-range {}'.format(

4
tools/ci_build/github/azure-pipelines/orttraining-pai-ci-pipeline.yml
View file

@ -12,8 +12,8 @@ variables:
# generated from tools/ci_build/github/pai/rocm-ci-pipeline-env.Dockerfile
container:
image: onnxruntimebuildcache.azurecr.io/rocm-ci-pipeline-env
endpoint: onnxruntimebuildcache
image: onnxruntimecibuildenvironment.azurecr.io/rocm-ci-pipeline-env
endpoint: onnxruntimecibuildenvironmentforamd
options: --privileged -e HIP_VISIBLE_DEVICES --security-opt seccomp=unconfined --device=/dev/kfd --device=/dev/dri --group-add $(video) --group-add $(render)
steps: