Skip to content

A
akg

MindSpore / akg

通知 58
Star 7
Fork 7
A
akg
提交 569a92fa 编写于 作者: L looop5
浏览文件
操作

fix CI matmul run error

上级 a8d1a104
隐藏空白更改
内联 并排
Showing with 3 addition and 3 deletion
tests/operators/cube/test_matmul_001.py
浏览文件 @ 569a92fa
......@@ -36,13 +36,13 @@ class TestCase(TestBase):
# bert shape with batch
# [16, 16, 32, 32, 16, 16] * [16, 16, 4, 32, 16, 16] -> [16, 16, 4, 32, 16, 16]
("batchmatmul_run_bert_00", "matmul_run", ((16, 16, 512, 512), (16, 16, 512, 64), 0,
"zN", "zN", "zN", False, False, "float16", "float16", "batchmatmul_cce")),
"zN", "zN", "zN", False, False, "float16", None, "float16", "batchmatmul_cce")),
# [16, 16, 32, 32, 16, 16] * [16, 16, 4, 32, 16, 16] -> [16, 16, 4, 32, 16, 16]
("batchmatmul_run_bert_01", "matmul_run", ((16, 16, 512, 512), (16, 16, 512, 64), 0,
"zN", "zN", "zN", True, False, "float16", "float16", "batchmatmul_cce")),
"zN", "zN", "zN", True, False, "float16", None, "float16", "batchmatmul_cce")),
# [16, 16, 4, 32, 16, 16] * [16, 16, 4, 32, 16, 16] -> [16, 16, 32, 32, 16, 16]
("batchmatmul_run_bert_02", "matmul_run", ((16, 16, 512, 64), (16, 16, 512, 64), 0,
"zN", "zN", "zN", False, True, "float16", "float16", "batchmatmul_cce")),
"zN", "zN", "zN", False, True, "float16", None, "float16", "batchmatmul_cce")),
# bert shape
("matmul_run_bert_00", "matmul_run", ((16, 1024), (16, 1024), 0, "zN", "zN", "zN", False, True, "float16", None, "float16", "matmul_cce")),
("matmul_run_bert_01", "matmul_run", ((8192, 4096), (8192, 1024), 0, "zN", "zN", "zN", True, False, "float16", None, "float32", "matmul_cce")),
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或