From 3a254b7ea65a6772bf1f185407f61b22e449f0e1 Mon Sep 17 00:00:00 2001 From: wangzhuo325 Date: Wed, 24 Jun 2020 14:58:31 +0800 Subject: [PATCH] add tilling config for batchmatmul --- python/akg/ops/nn/matmul.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/python/akg/ops/nn/matmul.py b/python/akg/ops/nn/matmul.py index 4830cd5..7e5cfd9 100644 --- a/python/akg/ops/nn/matmul.py +++ b/python/akg/ops/nn/matmul.py @@ -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}), } -- GitLab