Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MindSpore
akg
提交
5a919449
A
akg
项目概览
MindSpore
/
akg
通知
58
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看板
提交
5a919449
编写于
7月 07, 2020
作者:
L
looop5
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix variable broadcast_idx redefinition error when pragma dma_copy is replaced by opt_broadcast
上级
00df47a0
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
22 addition
and
15 deletion
+22
-15
src/emit_insn/insn_emitter_multimask.cc
src/emit_insn/insn_emitter_multimask.cc
+15
-9
src/pass/optimize_pragma.cc
src/pass/optimize_pragma.cc
+7
-6
未找到文件。
src/emit_insn/insn_emitter_multimask.cc
浏览文件 @
5a919449
...
...
@@ -86,19 +86,19 @@ Stmt MultiMaskEmitter(const Stmt &stmt) {
int
i_loop_elements
=
LeastCommonMultiple
(
broadcast_len
,
block_size
);
int
mask_num
=
i_loop_elements
/
broadcast_len
;
auto
i_var
=
VarExpr
(
"broadcast_idx"
);
Stmt
body
=
Evaluate
::
make
(
0
);
for
(
int
i
=
0
;
i
<
std
::
min
(
mask_num
,
data_len
);
i
++
)
{
auto
i_var
=
VarExpr
(
"broadcast_idx"
+
std
::
to_string
(
i
));
auto
dst_block_offset
=
(
i
*
broadcast_len
)
%
block_size
;
auto
dst_block_cnt
=
(
i
*
broadcast_len
)
-
dst_block_offset
;
Expr
base_addr_offset
=
i_var
*
i_loop_elements
+
dst_block_cnt
;
auto
base_src
=
src_buffer_id
.
vload
({
i_var
*
mask_num
+
i
},
i_type
);
CHECK_NE
(
i_loop_elements
,
0
);
auto
loop_num
=
(
data_len
+
mask_num
-
1
-
i
)
*
broadcast_len
/
i_loop_elements
;
// GenHead
Expr
base_addr_offset
=
i_var
*
i_loop_elements
+
dst_block_cnt
;
auto
base_src
=
src_buffer_id
.
vload
({
i_var
*
mask_num
+
i
},
i_type
);
int
head_size
=
std
::
min
(
simd_size
-
dst_block_offset
,
broadcast_len
);
auto
vec_mask_head
=
GetVecMaskWithOffset
(
head_size
,
dst_block_offset
,
i_type
);
auto
head_mask
=
EmitSetVecMaskIntrin
(
Stmt
(),
i_type
,
vec_mask_head
);
...
...
@@ -111,30 +111,36 @@ Stmt MultiMaskEmitter(const Stmt &stmt) {
// GenBody
if
(
dst_block_offset
+
broadcast_len
>=
simd_size
*
2
)
{
auto
i_var_body
=
VarExpr
(
i_var
->
name_hint
+
"_body"
);
Expr
base_addr_offset_body
=
i_var_body
*
i_loop_elements
+
dst_block_cnt
;
auto
base_src_body
=
src_buffer_id
.
vload
({
i_var_body
*
mask_num
+
i
},
i_type
);
int
repeat_size
=
(
dst_block_offset
+
broadcast_len
)
/
simd_size
-
1
;
auto
vec_mask_body
=
GetVecMaskWithOffset
(
simd_size
,
0
,
i_type
);
auto
full_mask
=
EmitSetVecMaskIntrin
(
Stmt
(),
i_type
,
vec_mask_body
);
int
body_addr_offset
=
simd_size
;
Expr
body_dst
=
GetAccessPtr
(
dst_buffer_id
,
"w"
,
base_addr_offset
+
body_addr_offset
);
Expr
body_dst
=
GetAccessPtr
(
dst_buffer_id
,
"w"
,
base_addr_offset
_body
+
body_addr_offset
);
auto
body_dump
=
Evaluate
::
make
(
Call
::
make
(
i_type
,
"vector_dup"
,
{
body_dst
,
base_src
,
repeat_size
,
1
,
1
,
8
,
8
},
Call
::
Extern
));
Evaluate
::
make
(
Call
::
make
(
i_type
,
"vector_dup"
,
{
body_dst
,
base_src
_body
,
repeat_size
,
1
,
1
,
8
,
8
},
Call
::
Extern
));
auto
body_gen
=
Block
::
make
({
full_mask
,
body_dump
});
auto
body_stmt
=
For
::
make
(
i_var
,
Expr
(
0
),
Expr
(
loop_num
),
ForType
::
Serial
,
DeviceAPI
::
None
,
body_gen
);
auto
body_stmt
=
For
::
make
(
i_var
_body
,
Expr
(
0
),
Expr
(
loop_num
),
ForType
::
Serial
,
DeviceAPI
::
None
,
body_gen
);
ret_stmt
=
Block
::
make
(
ret_stmt
,
body_stmt
);
}
// GenTail
if
((
dst_block_offset
+
broadcast_len
)
%
simd_size
!=
0
&&
dst_block_offset
+
broadcast_len
>
simd_size
)
{
auto
i_var_tail
=
VarExpr
(
i_var
->
name_hint
+
"_tail"
);
Expr
base_addr_offset_body
=
i_var_tail
*
i_loop_elements
+
dst_block_cnt
;
auto
base_src_body
=
src_buffer_id
.
vload
({
i_var_tail
*
mask_num
+
i
},
i_type
);
int
tail_size
=
(
dst_block_offset
+
broadcast_len
)
%
simd_size
;
auto
vec_mask_tail
=
GetVecMaskWithOffset
(
tail_size
,
0
,
i_type
);
auto
tail_mask
=
EmitSetVecMaskIntrin
(
Stmt
(),
i_type
,
vec_mask_tail
);
int
tail_addr_offset
=
dst_block_offset
+
broadcast_len
-
tail_size
;
Expr
tail_dst
=
GetAccessPtr
(
dst_buffer_id
,
"w"
,
base_addr_offset
+
tail_addr_offset
);
Expr
tail_dst
=
GetAccessPtr
(
dst_buffer_id
,
"w"
,
base_addr_offset
_body
+
tail_addr_offset
);
auto
tail_dump
=
Evaluate
::
make
(
Call
::
make
(
i_type
,
"vector_dup"
,
{
tail_dst
,
base_src
,
1
,
1
,
1
,
1
,
1
},
Call
::
Extern
));
Evaluate
::
make
(
Call
::
make
(
i_type
,
"vector_dup"
,
{
tail_dst
,
base_src
_body
,
1
,
1
,
1
,
1
,
1
},
Call
::
Extern
));
auto
tail
=
Block
::
make
({
tail_mask
,
tail_dump
});
auto
tail_stmt
=
For
::
make
(
i_var
,
Expr
(
0
),
Expr
(
loop_num
),
ForType
::
Serial
,
DeviceAPI
::
None
,
tail
);
auto
tail_stmt
=
For
::
make
(
i_var
_tail
,
Expr
(
0
),
Expr
(
loop_num
),
ForType
::
Serial
,
DeviceAPI
::
None
,
tail
);
ret_stmt
=
Block
::
make
(
ret_stmt
,
tail_stmt
);
}
...
...
src/pass/optimize_pragma.cc
浏览文件 @
5a919449
...
...
@@ -138,7 +138,11 @@ class EstimateAlign : public IRMutator {
public:
bool
IsSimpleAddress
(
const
Stmt
&
stmt
)
{
Mutate
(
stmt
);
return
all_simple_addressing_
;
// Returns true only when the numbers of Store in IR that is not elementwise
// is only 1 or less, in this case, we can consider optimizing broadcast by
// using variable length mask in insn emitting pass safely because at most
// 1 Store does not need to cosider block alignment.
return
(
not_simple_addressing_cnt_
<
2
);
}
Stmt
Mutate_
(
const
AttrStmt
*
op
,
const
Stmt
&
stmt
)
final
{
...
...
@@ -146,22 +150,19 @@ class EstimateAlign : public IRMutator {
if
(
exclude_list
.
count
(
op
->
value
.
as
<
StringImm
>
()
->
value
))
{
return
stmt
;
}
is_candidate_
=
true
;
StmtInfoList
dst_info_list
,
src_info_list
;
StmtInfo
if_info
,
for_info
;
GetCompactComputationInfo
(
op
->
body
,
dst_info_list
,
src_info_list
,
if_info
,
for_info
,
false
);
if
(
!
src_info_list
.
empty
()
&&
!
IsElementwise
(
dst_info_list
,
src_info_list
))
{
all_simple_addressing_
=
false
;
not_simple_addressing_cnt_
++
;
}
is_candidate_
=
false
;
}
return
IRMutator
::
Mutate_
(
op
,
stmt
);
}
bool
is_candidate_
{
false
};
bool
all_simple_addressing_
{
true
};
int
not_simple_addressing_cnt_
{
0
};
// records the number of stores that are not elementwise
};
Stmt
OptimizePragma
(
Stmt
stmt
)
{
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录