Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Guile_
akg
提交
2fb61524
A
akg
项目概览
Guile_
/
akg
与 Fork 源项目一致
Fork自
MindSpore / akg
通知
1
Star
0
Fork
0
代码
文件
提交
分支
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看板
体验新版 GitCode,发现更多精彩内容 >>
提交
2fb61524
编写于
6月 23, 2020
作者:
M
mindspore-ci-bot
提交者:
Gitee
6月 23, 2020
浏览文件
操作
浏览文件
下载
差异文件
!12 matmul op input tensor fixed
Merge pull request !12 from hiranoaya/format
上级
76dcec1c
dc4be2ad
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
22 addition
and
22 deletion
+22
-22
python/akg/ops/nn/matmul.py
python/akg/ops/nn/matmul.py
+22
-22
未找到文件。
python/akg/ops/nn/matmul.py
浏览文件 @
2fb61524
...
@@ -33,51 +33,51 @@ matmul_set_dim_map = {
...
@@ -33,51 +33,51 @@ matmul_set_dim_map = {
# bert best tile
# bert best tile
# (16, 1024), (16, 1024)
# (16, 1024), (16, 1024)
str
(((
1
,
64
,
1
,
16
,
16
),
(
1
,
64
,
1
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
True
,
"float16"
,
"float16"
))
:
([(
16
,
16
),(
16
,
16
),(
32
,
32
)],
{
"bypass"
:
2
}),
str
(((
64
,
1
,
16
,
16
),
(
64
,
1
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
True
,
"float16"
,
"float16"
))
:
([(
16
,
16
),(
16
,
16
),(
32
,
32
)],
{
"bypass"
:
2
}),
# (8192, 4096), (8192, 1024)
# (8192, 4096), (8192, 1024)
str
(((
1
,
256
,
512
,
16
,
16
),
(
1
,
64
,
512
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
True
,
False
,
"float16"
,
"float32"
))
:
([(
8
,
8
),(
16
,
16
),(
16
,
16
),(
16
,
16
),(
8
,
1
)],
{
"bypass"
:
0
}),
str
(((
256
,
512
,
16
,
16
),
(
64
,
512
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
True
,
False
,
"float16"
,
"float32"
))
:
([(
8
,
8
),(
16
,
16
),(
16
,
16
),(
16
,
16
),(
8
,
1
)],
{
"bypass"
:
0
}),
# (8192, 1024), (1024, 4096)
# (8192, 1024), (1024, 4096)
str
(((
1
,
64
,
512
,
16
,
16
),
(
1
,
256
,
64
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
False
,
"float16"
,
"float16"
))
:
([(
16
,
16
),(
8
,
4
),(
16
,
16
),(
16
,
16
),(
64
,
8
)],
{
"bypass"
:
0
}),
str
(((
64
,
512
,
16
,
16
),
(
256
,
64
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
False
,
"float16"
,
"float16"
))
:
([(
16
,
16
),(
8
,
4
),(
16
,
16
),(
16
,
16
),(
64
,
8
)],
{
"bypass"
:
0
}),
# (16, 16), (16, 1024)
# (16, 16), (16, 1024)
str
(((
1
,
1
,
1
,
16
,
16
),
(
1
,
64
,
1
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
True
,
False
,
"float16"
,
"float32"
))
:
([(
8
,
8
),(
16
,
16
),(
16
,
16
),(
16
,
16
)],
{
"bypass"
:
0
}),
str
(((
1
,
1
,
1
6
,
16
),
(
64
,
1
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
True
,
False
,
"float16"
,
"float32"
))
:
([(
8
,
8
),(
16
,
16
),(
16
,
16
),(
16
,
16
)],
{
"bypass"
:
0
}),
# (1216, 1024), (1024, 1024)
# (1216, 1024), (1024, 1024)
str
(((
1
,
64
,
76
,
16
,
16
),
(
1
,
64
,
64
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
False
,
"float16"
,
"float32"
))
:
([(
4
,
4
),(
19
,
19
),(
16
,
16
),(
16
,
16
),(
4
,
1
)],
{
"bypass"
:
0
}),
str
(((
64
,
76
,
16
,
16
),
(
64
,
64
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
False
,
"float16"
,
"float32"
))
:
([(
4
,
4
),(
19
,
19
),(
16
,
16
),(
16
,
16
),(
4
,
1
)],
{
"bypass"
:
0
}),
# (8192, 4096), (4096, 1024)
# (8192, 4096), (4096, 1024)
str
(((
1
,
256
,
512
,
16
,
16
),
(
1
,
64
,
256
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
False
,
"float16"
,
"float16"
))
:
([(
8
,
8
),(
32
,
32
),(
16
,
16
),(
16
,
16
),(
2
,
1
)],
{
"bypass"
:
0
}),
str
(((
256
,
512
,
16
,
16
),
(
64
,
256
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
False
,
"float16"
,
"float16"
))
:
([(
8
,
8
),(
32
,
32
),(
16
,
16
),(
16
,
16
),(
2
,
1
)],
{
"bypass"
:
0
}),
# (8192, 1024), (4096, 1024)
# (8192, 1024), (4096, 1024)
str
(((
1
,
64
,
512
,
16
,
16
),
(
1
,
64
,
256
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
True
,
"float16"
,
"float16"
))
:
([(
8
,
8
),(
32
,
32
),(
16
,
16
),(
16
,
16
),(
2
,
1
)],
{
"bypass"
:
0
}),
str
(((
64
,
512
,
16
,
16
),
(
64
,
256
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
True
,
"float16"
,
"float16"
))
:
([(
8
,
8
),(
32
,
32
),(
16
,
16
),(
16
,
16
),(
2
,
1
)],
{
"bypass"
:
0
}),
# (8192, 1024), (8192, 4096)
# (8192, 1024), (8192, 4096)
str
(((
1
,
64
,
512
,
16
,
16
),
(
1
,
256
,
512
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
True
,
False
,
"float16"
,
"float16"
))
:
([[
8
,
8
],
[
32
,
32
],
[
16
,
16
],
[
16
,
16
],
[
16
,
2
]],
{
"bypass"
:
0
}),
str
(((
64
,
512
,
16
,
16
),
(
256
,
512
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
True
,
False
,
"float16"
,
"float16"
))
:
([[
8
,
8
],
[
32
,
32
],
[
16
,
16
],
[
16
,
16
],
[
16
,
2
]],
{
"bypass"
:
0
}),
# (1216, 1024), (1024, 1024)
# (1216, 1024), (1024, 1024)
str
(((
1
,
64
,
76
,
16
,
16
),
(
1
,
64
,
64
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
True
,
"float16"
,
"float16"
))
:
([(
8
,
8
),(
19
,
19
),(
16
,
16
),(
16
,
16
),(
16
,
1
)],
{
"bypass"
:
2
}),
str
(((
64
,
76
,
16
,
16
),
(
64
,
64
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
True
,
"float16"
,
"float16"
))
:
([(
8
,
8
),(
19
,
19
),(
16
,
16
),(
16
,
16
),(
16
,
1
)],
{
"bypass"
:
2
}),
# (8192, 1024), (1024, 1024)
# (8192, 1024), (1024, 1024)
str
(((
1
,
64
,
512
,
16
,
16
),
(
1
,
64
,
64
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
False
,
"float16"
,
"float16"
))
:
([(
16
,
4
),(
16
,
8
),(
16
,
16
),(
16
,
16
),(
64
,
16
)],
{
"bypass"
:
0
}),
str
(((
64
,
512
,
16
,
16
),
(
64
,
64
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
False
,
"float16"
,
"float16"
))
:
([(
16
,
4
),(
16
,
8
),(
16
,
16
),(
16
,
16
),(
64
,
16
)],
{
"bypass"
:
0
}),
# (1216, 30522), (30522, 1024)
# (1216, 30522), (30522, 1024)
str
(((
1
,
1908
,
76
,
16
,
16
),
(
1
,
64
,
1908
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
False
,
"float16"
,
"float16"
))
:
([(
8
,
8
),(
19
,
19
),(
16
,
16
),(
16
,
16
),(
6
,
1
)],
{
"bypass"
:
0
}),
str
(((
1
908
,
76
,
16
,
16
),
(
64
,
1908
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
False
,
"float16"
,
"float16"
))
:
([(
8
,
8
),(
19
,
19
),(
16
,
16
),(
16
,
16
),(
6
,
1
)],
{
"bypass"
:
0
}),
# (1216, 30522), (1216, 1024)
# (1216, 30522), (1216, 1024)
str
(((
1
,
1908
,
76
,
16
,
16
),
(
1
,
64
,
76
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
True
,
False
,
"float16"
,
"float32"
))
:
([(
4
,
4
),(
18
,
18
),(
16
,
16
),(
16
,
16
),(
2
,
2
)],
{
"bypass"
:
0
}),
str
(((
1
908
,
76
,
16
,
16
),
(
64
,
76
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
True
,
False
,
"float16"
,
"float32"
))
:
([(
4
,
4
),(
18
,
18
),(
16
,
16
),(
16
,
16
),(
2
,
2
)],
{
"bypass"
:
0
}),
# (1216, 1024), (30522, 1024)
# (1216, 1024), (30522, 1024)
str
(((
1
,
64
,
76
,
16
,
16
),
(
1
,
64
,
1908
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
True
,
"float16"
,
"float32"
))
:
([(
9
,
9
),(
19
,
19
),(
16
,
16
),(
16
,
16
),(
64
,
1
)],
{
"bypass"
:
0
}),
str
(((
64
,
76
,
16
,
16
),
(
64
,
1908
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
True
,
"float16"
,
"float32"
))
:
([(
9
,
9
),(
19
,
19
),(
16
,
16
),(
16
,
16
),(
64
,
1
)],
{
"bypass"
:
0
}),
# (8192, 1024), (8192, 1024)
# (8192, 1024), (8192, 1024)
str
(((
1
,
64
,
512
,
16
,
16
),
(
1
,
64
,
512
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
True
,
False
,
"float16"
,
"float32"
))
:
([(
4
,
4
),(
16
,
16
),(
16
,
16
),(
16
,
16
),(
16
,
4
)],
{
"bypass"
:
0
}),
str
(((
64
,
512
,
16
,
16
),
(
64
,
512
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
True
,
False
,
"float16"
,
"float32"
))
:
([(
4
,
4
),(
16
,
16
),(
16
,
16
),(
16
,
16
),(
16
,
4
)],
{
"bypass"
:
0
}),
# (1216, 1024), (1216, 1024)
# (1216, 1024), (1216, 1024)
str
(((
1
,
64
,
76
,
16
,
16
),
(
1
,
64
,
76
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
True
,
False
,
"float16"
,
"float16"
))
:
([(
16
,
16
),(
8
,
8
),(
16
,
16
),(
16
,
16
),(
4
,
2
)],
{
"bypass"
:
0
}),
str
(((
64
,
76
,
16
,
16
),
(
64
,
76
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
True
,
False
,
"float16"
,
"float16"
))
:
([(
16
,
16
),(
8
,
8
),(
16
,
16
),(
16
,
16
),(
4
,
2
)],
{
"bypass"
:
0
}),
# (16, 1024), (16, 1024)
# (16, 1024), (16, 1024)
str
(((
1
,
64
,
1
,
16
,
16
),
(
1
,
64
,
1
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
True
,
False
,
"float16"
,
"float32"
))
:
([(
8
,
8
),(
2
,
2
),(
16
,
16
),(
16
,
16
),(
16
,
16
)],
{
"bypass"
:
0
}),
str
(((
64
,
1
,
16
,
16
),
(
64
,
1
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
True
,
False
,
"float16"
,
"float32"
))
:
([(
8
,
8
),(
2
,
2
),(
16
,
16
),(
16
,
16
),(
16
,
16
)],
{
"bypass"
:
0
}),
# (16, 1024), (1024, 1024)
# (16, 1024), (1024, 1024)
str
(((
1
,
64
,
1
,
16
,
16
),
(
1
,
64
,
64
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
True
,
"float16"
,
"float32"
))
:
([(
8
,
8
),(
16
,
16
),(
16
,
16
),(
32
,
8
)],
{
"bypass"
:
2
}),
str
(((
64
,
1
,
16
,
16
),
(
64
,
64
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
True
,
"float16"
,
"float32"
))
:
([(
8
,
8
),(
16
,
16
),(
16
,
16
),(
32
,
8
)],
{
"bypass"
:
2
}),
# (16, 16), (16, 1024)
# (16, 16), (16, 1024)
str
(((
1
,
1
,
1
,
16
,
16
),
(
1
,
64
,
1
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
False
,
"float16"
,
"float32"
))
:
([(
8
,
8
),(
16
,
16
),(
16
,
16
),(
16
,
16
)],
{
"bypass"
:
0
}),
str
(((
1
,
1
,
1
6
,
16
),
(
64
,
1
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
False
,
"float16"
,
"float32"
))
:
([(
8
,
8
),(
16
,
16
),(
16
,
16
),(
16
,
16
)],
{
"bypass"
:
0
}),
# (8192, 1024), (1024, 1024)
# (8192, 1024), (1024, 1024)
str
(((
1
,
64
,
512
,
16
,
16
),
(
1
,
64
,
64
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
True
,
"float16"
,
"float16"
))
:
([(
16
,
8
),(
8
,
8
),(
16
,
16
),(
16
,
16
),(
64
,
8
)],
{
"bypass"
:
1
}),
str
(((
64
,
512
,
16
,
16
),
(
64
,
64
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
True
,
"float16"
,
"float16"
))
:
([(
16
,
8
),(
8
,
8
),(
16
,
16
),(
16
,
16
),(
64
,
8
)],
{
"bypass"
:
1
}),
# (8192, 4096), (1024, 4096)
# (8192, 4096), (1024, 4096)
str
(((
1
,
256
,
512
,
16
,
16
),
(
1
,
256
,
64
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
True
,
"float16"
,
"float16"
))
:
([(
8
,
8
),(
16
,
16
),(
16
,
16
),(
16
,
16
),(
128
,
8
)],
{
"bypass"
:
1
}),
str
(((
256
,
512
,
16
,
16
),
(
256
,
64
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
True
,
"float16"
,
"float16"
))
:
([(
8
,
8
),(
32
,
32
),(
16
,
16
),(
16
,
16
),(
1
,
1
)],
{
"bypass"
:
0
}),
}
}
def
matmul_set_dim
(
A
,
B
,
b
,
out_dtype
,
left_format
,
right_format
,
output_format
,
adj_x
,
adj_y
,
attrs
):
def
matmul_set_dim
(
A
,
B
,
b
,
out_dtype
,
left_format
,
right_format
,
output_format
,
adj_x
,
adj_y
,
attrs
):
shape_A
=
A
.
shape
shape_A
=
A
.
shape
[
1
:
5
]
if
len
(
A
.
shape
)
==
5
else
A
.
shape
shape_B
=
B
.
shape
shape_B
=
B
.
shape
[
1
:
5
]
if
len
(
B
.
shape
)
==
5
else
B
.
shape
bias
=
0
if
b
is
None
else
1
bias
=
0
if
b
is
None
else
1
key
=
()
key
=
()
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录