Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MindSpore
akg
提交
3a254b7e
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看板
提交
3a254b7e
编写于
6月 24, 2020
作者:
W
wangzhuo325
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add tilling config for batchmatmul
上级
843b6771
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
5 addition
and
0 deletion
+5
-0
python/akg/ops/nn/matmul.py
python/akg/ops/nn/matmul.py
+5
-0
未找到文件。
python/akg/ops/nn/matmul.py
浏览文件 @
3a254b7e
...
...
@@ -72,6 +72,11 @@ matmul_set_dim_map = {
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)
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
}),
# (16, 16, 512, 512), (16, 16, 512, 64)
str
(((
16
,
16
,
32
,
32
,
16
,
16
),
(
16
,
16
,
4
,
32
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
False
,
"float16"
,
"float16"
))
:
([(
1
,
1
),(
1
,
1
),(
4
,
4
),(
16
,
16
),(
16
,
16
),(
16
,
16
),(
16
,
4
)],
{
"bypass"
:
0
}),
str
(((
16
,
16
,
32
,
32
,
16
,
16
),
(
16
,
16
,
4
,
32
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
True
,
False
,
"float16"
,
"float16"
))
:
([(
1
,
1
),(
1
,
1
),(
4
,
4
),(
16
,
16
),(
16
,
16
),(
16
,
16
),(
16
,
4
)],
{
"bypass"
:
0
}),
# (16, 16, 512, 64), (16, 16, 512, 64)
str
(((
16
,
16
,
4
,
32
,
16
,
16
),
(
16
,
16
,
4
,
32
,
16
,
16
),
0
,
"zN"
,
"zN"
,
"zN"
,
False
,
True
,
"float16"
,
"float16"
))
:
([(
1
,
1
),(
1
,
1
),(
32
,
4
),(
32
,
32
),(
16
,
16
),(
16
,
16
),(
4
,
4
)],
{
"bypass"
:
0
}),
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录