Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MindSpore
akg
提交
4cf8dda4
A
akg
项目概览
MindSpore
/
akg
通知
59
Star
7
Fork
7
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
A
akg
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
4cf8dda4
编写于
7月 25, 2020
作者:
D
dabaiji
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix wrong hoist problem in multicore loop switch hoist pass
上级
3b74f272
变更
1
显示空白变更内容
内联
并排
Showing
1 changed file
with
9 addition
and
1 deletion
+9
-1
src/pass/multi_core_loop_switch_hoist.cc
src/pass/multi_core_loop_switch_hoist.cc
+9
-1
未找到文件。
src/pass/multi_core_loop_switch_hoist.cc
浏览文件 @
4cf8dda4
...
...
@@ -176,7 +176,7 @@ class MultiCoreLoopHoister : public IRMutator {
}
Stmt
Mutate_
(
const
IfThenElse
*
op
,
const
Stmt
&
s
)
final
{
if
(
loop_stack_
.
empty
()
||
!
need_hoisted_
)
{
if
(
loop_stack_
.
empty
()
||
!
need_hoisted_
||
IsScalar
(
op
->
then_case
)
)
{
return
IRMutator
::
Mutate_
(
op
,
s
);
}
loop_stack_
.
back
().
branch_bound
=
true
;
...
...
@@ -298,6 +298,14 @@ class MultiCoreLoopHoister : public IRMutator {
return
false
;
}
bool
IsScalar
(
const
Stmt
&
s
)
{
if
(
s
->
IsInstance
<
Store
>
())
{
auto
store
=
s
.
as
<
Store
>
();
return
IsConstExpr
(
store
->
predicate
);
}
return
false
;
}
struct
LoopEntry
{
const
For
*
node
;
bool
branch_bound
;
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录