From 76f610b3ed69075e85b68f72680d0e3812d3aeef Mon Sep 17 00:00:00 2001 From: dingpeifei Date: Thu, 22 Apr 2021 11:43:58 +0800 Subject: [PATCH] upgrade_ascend_0422 --- graphengine | 2 +- mindspore/_check_version.py | 2 +- mindspore/_extends/parallel_compile/tbe_compiler/compiler.py | 1 + 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/graphengine b/graphengine index 8770bfcdd73..708f523f053 160000 --- a/graphengine +++ b/graphengine @@ -1 +1 @@ -Subproject commit 8770bfcdd73777207d562597e21c63179af598f2 +Subproject commit 708f523f05389c152a429ecbba9782e0b93fe42b diff --git a/mindspore/_check_version.py b/mindspore/_check_version.py index 978d748c2be..66a53916678 100644 --- a/mindspore/_check_version.py +++ b/mindspore/_check_version.py @@ -200,7 +200,7 @@ class AscendEnvChecker(EnvChecker): """ascend environment check""" def __init__(self): - self.version = ["1.77.22.6.220"] + self.version = ["1.78.T2.0.B020"] atlas_nnae_version = "/usr/local/Ascend/nnae/latest/fwkacllib/version.info" atlas_toolkit_version = "/usr/local/Ascend/ascend-toolkit/latest/fwkacllib/version.info" hisi_fwk_version = "/usr/local/Ascend/fwkacllib/version.info" diff --git a/mindspore/_extends/parallel_compile/tbe_compiler/compiler.py b/mindspore/_extends/parallel_compile/tbe_compiler/compiler.py index 118ed80d28e..ba8c32c61ae 100755 --- a/mindspore/_extends/parallel_compile/tbe_compiler/compiler.py +++ b/mindspore/_extends/parallel_compile/tbe_compiler/compiler.py @@ -148,6 +148,7 @@ def compile_fusion_op(json_str): te_set_version(args['fusion_op']["socVersion"]) if 'fusion_op' not in args or not args['fusion_op']: raise ValueError("Json string Errors, key:fusion_op not found.") + args['fusion_op']['SocInfo'] = args['SocInfo'] fusion_op_arg = args['fusion_op'] return fusion_op(json.dumps(fusion_op_arg))