!9735 fix st failed occasionally

From: @jojobugfree
Reviewed-by: @zhoufeng54,@chujinjin
Signed-off-by: @chujinjin
pull/9735/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 6439f676a9

@ -51,11 +51,11 @@ def change_current_dump_json(file_name, dump_path):
def test_async_dump(): def test_async_dump():
context.set_context(mode=context.GRAPH_MODE, device_target="Ascend") context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
pwd = os.getcwd() pwd = os.getcwd()
dump_path = pwd + "/dump" dump_path = pwd + "/async_dump"
change_current_dump_json('async_dump.json', dump_path) change_current_dump_json('async_dump.json', dump_path)
os.environ['MINDSPORE_DUMP_CONFIG'] = pwd + "/async_dump.json" os.environ['MINDSPORE_DUMP_CONFIG'] = pwd + "/async_dump.json"
device_id = context.get_context("device_id") device_id = context.get_context("device_id")
dump_file_path = pwd + '/dump/device_{}/Net_graph_0/0/0/'.format(device_id) dump_file_path = pwd + '/async_dump/device_{}/Net_graph_0/0/0/'.format(device_id)
if os.path.isdir(dump_path): if os.path.isdir(dump_path):
shutil.rmtree(dump_path) shutil.rmtree(dump_path)
add = Net() add = Net()
@ -70,11 +70,11 @@ def test_async_dump():
def test_e2e_dump(): def test_e2e_dump():
context.set_context(mode=context.GRAPH_MODE, device_target="Ascend") context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
pwd = os.getcwd() pwd = os.getcwd()
dump_path = pwd + "/dump" dump_path = pwd + "/e2e_dump"
change_current_dump_json('e2e_dump.json', dump_path) change_current_dump_json('e2e_dump.json', dump_path)
os.environ['MINDSPORE_DUMP_CONFIG'] = pwd + "/e2e_dump.json" os.environ['MINDSPORE_DUMP_CONFIG'] = pwd + "/e2e_dump.json"
device_id = context.get_context("device_id") device_id = context.get_context("device_id")
dump_file_path = pwd + '/dump/Net/device_{}/iteration_1/'.format(device_id) dump_file_path = pwd + '/e2e_dump/Net/device_{}/iteration_1/'.format(device_id)
if os.path.isdir(dump_path): if os.path.isdir(dump_path):
shutil.rmtree(dump_path) shutil.rmtree(dump_path)
add = Net() add = Net()

Loading…
Cancel
Save