This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
AI开发
Register
Sign In
Huawei_Technology
/
mindspore
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
13
Wiki
evaluate
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
HPC
Browse Source
incoroprate TupleGetItem Switch if EnvSetItem or EnvGetItem exists in the same func graph, so EnvGetItem or EnvSetItem can be eliminated even DEFER_INLINE flag is set, so before second order J is expanded, EnvGetItem or EnvSetItem is eliminated
tags/v1.2.0-rc1
zhousiyi
5 years ago
parent
0bd1e34a4d
commit
3f2a08c1d0
1 changed files
with
2 additions
and
1 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+2
-1
mindspore/ccsrc/frontend/optimizer/irpass/incorporate_getitem.h
+ 2
- 1
mindspore/ccsrc/frontend/optimizer/irpass/incorporate_getitem.h
View File
@@ -362,7 +362,8 @@ class IncorporateGetitemSwitch : public AnfVisitor {
is_in_get_ = false;
auto fg = node->func_graph();
if (idx_ == -1 || switch_ == nullptr || fg == nullptr || fg->has_flag(FUNC_GRAPH_FLAG_DEFER_INLINE)) {
if (idx_ == -1 || switch_ == nullptr || fg == nullptr ||
(fg->has_flag(FUNC_GRAPH_FLAG_DEFER_INLINE) && !ExistEnvNode(fg))) {
return nullptr;
}
Write
Preview
Loading…
Cancel
Save