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

Fix for AMD GPU pipeline (#16357)

Browse source
This commit is contained in:
Baiju Meswani 2023-06-14 20:36:16 -07:00 committed by GitHub
parent 73dad4452b
commit 5eec24837f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -239,8 +239,7 @@ jobs:
python -m onnxruntime.training.ortmodule.torch_cpp_extensions.install; \
python orttraining_ortmodule_tests.py \
--mnist /mnist \
--bert_data /bert_data/hf_data/glue_data/CoLA/original/raw \
--transformers_cache /hf_models_cache/huggingface/transformers"
--bert_data /bert_data/hf_data/glue_data/CoLA/original/raw"
workingDirectory: $(Build.SourcesDirectory)
displayName: 'Run orttraining_ortmodule_tests.py'
condition: and(succeededOrFailed(), eq(variables.onnxruntimeBuildSucceeded, 'true'))