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

Re-enable test and increase timeout (#6785)

Browse source
This commit is contained in:
baijumeswani 2021-02-23 18:51:06 -08:00 committed by GitHub
parent 563218dcda
commit 65ba51d93e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
orttraining/orttraining/test/python/orttraining_ortmodule_tests.py
View file

@ -89,8 +89,7 @@ def main():
run_ort_module_hf_bert_for_sequence_classification_from_pretrained(cwd, log, no_cuda=False, data_dir=args.bert_data)
# TODO: Re-enable when hang with no_cuda=True is fixed
# run_ort_module_hf_bert_for_sequence_classification_from_pretrained(cwd, log, no_cuda=True, data_dir=args.bert_data)
run_ort_module_hf_bert_for_sequence_classification_from_pretrained(cwd, log, no_cuda=True, data_dir=args.bert_data)
return 0

2
tools/ci_build/github/azure-pipelines/orttraining-linux-gpu-ortmodule-test-ci-pipeline.yml
View file

@ -49,7 +49,7 @@ jobs:
--cwd /build/RelWithDebInfo \
displayName: 'Run orttraining_ortmodule_tests.py'
condition: succeededOrFailed()
timeoutInMinutes: 30
timeoutInMinutes: 60
- template: templates/component-governance-component-detection-steps.yml
parameters: