From 65ba51d93eed1e17d4b7ff48c98f571e5a6c34f2 Mon Sep 17 00:00:00 2001 From: baijumeswani Date: Tue, 23 Feb 2021 18:51:06 -0800 Subject: [PATCH] Re-enable test and increase timeout (#6785) --- .../orttraining/test/python/orttraining_ortmodule_tests.py | 3 +-- .../orttraining-linux-gpu-ortmodule-test-ci-pipeline.yml | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/orttraining/orttraining/test/python/orttraining_ortmodule_tests.py b/orttraining/orttraining/test/python/orttraining_ortmodule_tests.py index 83d52c75b3..3f6a268550 100644 --- a/orttraining/orttraining/test/python/orttraining_ortmodule_tests.py +++ b/orttraining/orttraining/test/python/orttraining_ortmodule_tests.py @@ -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 diff --git a/tools/ci_build/github/azure-pipelines/orttraining-linux-gpu-ortmodule-test-ci-pipeline.yml b/tools/ci_build/github/azure-pipelines/orttraining-linux-gpu-ortmodule-test-ci-pipeline.yml index 0d959046b0..1ee2de1a96 100644 --- a/tools/ci_build/github/azure-pipelines/orttraining-linux-gpu-ortmodule-test-ci-pipeline.yml +++ b/tools/ci_build/github/azure-pipelines/orttraining-linux-gpu-ortmodule-test-ci-pipeline.yml @@ -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: