From 467488ece94a5ebd18ae952c973237590392824c Mon Sep 17 00:00:00 2001 From: ckey_Dou Date: Tue, 23 Jun 2020 00:52:27 +0800 Subject: [PATCH] set tiling args by return value but not decorator --- python/akg/ops/nn/matmul.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/python/akg/ops/nn/matmul.py b/python/akg/ops/nn/matmul.py index b2e2369..4d74d27 100644 --- a/python/akg/ops/nn/matmul.py +++ b/python/akg/ops/nn/matmul.py @@ -299,8 +299,6 @@ def matmul4D_compute(x, y, bias_value, out_dtype, left_format, right_format, 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): """ 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=" out = matmul4D_compute(x, y, b, out_dtype, left_format, right_format, out_format, transpose_x, transpose_y, attrs) 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 -- GitLab