提交 843b6771 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!13 Update the way to set tiling args for matmul

Merge pull request !13 from chengbin/master
...@@ -299,8 +299,6 @@ def matmul4D_compute(x, y, bias_value, out_dtype, left_format, right_format, out ...@@ -299,8 +299,6 @@ def matmul4D_compute(x, y, bias_value, out_dtype, left_format, right_format, out
return out return out
@ct_util.reg_set_dim_func(matmul_set_dim)
def matmul(x, y, b, out_dtype, left_format="zZ", right_format="nZ", out_format="zN", transpose_x=False, transpose_y=False, attrs=None): def matmul(x, y, b, out_dtype, left_format="zZ", right_format="nZ", out_format="zN", transpose_x=False, transpose_y=False, attrs=None):
""" """
Computes matrix multiplication x * y + b. Computes matrix multiplication x * y + b.
...@@ -337,4 +335,8 @@ def matmul(x, y, b, out_dtype, left_format="zZ", right_format="nZ", out_format=" ...@@ -337,4 +335,8 @@ def matmul(x, y, b, out_dtype, left_format="zZ", right_format="nZ", out_format="
out = matmul4D_compute(x, y, b, out_dtype, left_format, right_format, out_format, transpose_x, transpose_y, attrs) out = matmul4D_compute(x, y, b, out_dtype, left_format, right_format, out_format, transpose_x, transpose_y, attrs)
attr_map = {"pragma_rmselfdep": False} attr_map = {"pragma_rmselfdep": False}
dims_info, _ = matmul_set_dim(x, y, b, out_dtype, left_format, right_format, out_format, transpose_x, transpose_y, attrs)
attr_map["dim"] = dims_info
return out, attr_map return out, attr_map
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册