upgrade_ascend_0422

This commit is contained in:
dingpeifei 2021-04-22 11:43:58 +08:00
parent 2a37201ea1
commit 76f610b3ed
3 changed files with 3 additions and 2 deletions

@ -1 +1 @@
Subproject commit 8770bfcdd73777207d562597e21c63179af598f2
Subproject commit 708f523f05389c152a429ecbba9782e0b93fe42b

View File

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

View File

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