!10897 Update graph kernel split model for Ascend

From: @gaoxiong1
Reviewed-by: @ryanww
Signed-off-by:
pull/10897/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit b41d83a7df

File diff suppressed because it is too large Load Diff

@ -25,8 +25,9 @@ def split_with_json(json_str: str):
"""Call costmodel to split GraphKernel"""
try:
graph_desc = json.loads(json_str)
target = graph_desc['process']
comp = model.load_composite(graph_desc)
graph_split, graph_mode = model.split(comp.graph)
graph_split, graph_mode = model.split(comp.graph, target)
is_multi_graph = len(graph_split) > 1
graph_list = list(map(comp.dump, graph_split))
result = {"multi_graph": is_multi_graph,

Loading…
Cancel
Save