From 37836f61f617d4aeb74c9c63ed004e5b53e8c61a Mon Sep 17 00:00:00 2001 From: caifubi Date: Wed, 9 Feb 2022 11:24:02 +0800 Subject: [PATCH] Fix ci pynative testcase failure --- tests/st/pynative/test_pynative_resnet50_ascend.py | 2 +- tests/st/pynative/test_pynative_resnet50_gpu.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/st/pynative/test_pynative_resnet50_ascend.py b/tests/st/pynative/test_pynative_resnet50_ascend.py index 2c95660afd6..287b83868fc 100644 --- a/tests/st/pynative/test_pynative_resnet50_ascend.py +++ b/tests/st/pynative/test_pynative_resnet50_ascend.py @@ -444,7 +444,7 @@ def test_pynative_resnet50(): def test_pynative_resnet50_with_env(queue, device_id, device_num): os.system("mkdir " + str(device_id)) os.chdir(str(device_id)) - context.set_context(mode=context.PYNATIVE_MODE, device_target="Ascend", device_id=device_id) + context.set_context(mode=context.PYNATIVE_MODE, device_target="Ascend", device_id=device_id, runtime_num_threads=20) os.environ['MINDSPORE_HCCL_CONFIG_PATH'] = MINDSPORE_HCCL_CONFIG_PATH os.environ['RANK_ID'] = str(device_id) os.environ['RANK_SIZE'] = str(device_num) diff --git a/tests/st/pynative/test_pynative_resnet50_gpu.py b/tests/st/pynative/test_pynative_resnet50_gpu.py index deb16c96076..0a21049d2b9 100644 --- a/tests/st/pynative/test_pynative_resnet50_gpu.py +++ b/tests/st/pynative/test_pynative_resnet50_gpu.py @@ -379,7 +379,7 @@ class GradWrap(Cell): def test_pynative_resnet50(): - context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU") + context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU", runtime_num_threads=20) context.reset_auto_parallel_context() context.set_auto_parallel_context(parallel_mode=ParallelMode.DATA_PARALLEL, gradients_mean=False, device_num=8) init()