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

!10938 Set default mode for output reshape op 

From: @gaoxiong1
Reviewed-by: @ryanww,@anyrenwei,@dylangeng
Signed-off-by: @anyrenwei,@dylangeng
This commit is contained in:
mindspore-ci-bot 2021-01-05 17:10:40 +08:00 committed by Gitee
parent 377796e7e4 cc89c3f896
commit 446b92407b
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/_extends/graph_kernel/model/graph_split.py
View File

@ -1,4 +1,4 @@
# Copyright 2020 Huawei Technologies Co., Ltd
# Copyright 2020-2021 Huawei Technologies Co., Ltd
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
@ -213,7 +213,9 @@ class GraphSplitByPattern:
break
if out_reshape_ops:
for op in out_reshape_ops:
new_areas.append(self.Area(op, False))
a = self.Area(op, False)
self.set_default_mode(a)
new_areas.append(a)
area.ops = remain_ops
if len(remain_ops) == 1:
self.set_default_mode(area)