p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!29813 PyNative 8P performance 

Merge pull request !29813 from caifubi/master-pynative-perfmance-testcase
This commit is contained in:
i-robot 2022-02-09 09:36:12 +00:00 committed by Gitee
parent 527a20b5f4 37836f61f6
commit 07823a918f
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/st/pynative/test_pynative_resnet50_ascend.py
View File

@ -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)

2
tests/st/pynative/test_pynative_resnet50_gpu.py
View File

@ -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()