From 32916b68e9a060ef2e7244a06c4f657586fffbec Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Thu, 31 Mar 2022 18:53:48 +0900 Subject: [PATCH] fixed pack_data --- python/tvm/topi/nn/conv2d.py | 1 - python/tvm/topi/x86/conv2d_int8.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/python/tvm/topi/nn/conv2d.py b/python/tvm/topi/nn/conv2d.py index 68eb4eb6f01b2..c27ea81144ac2 100644 --- a/python/tvm/topi/nn/conv2d.py +++ b/python/tvm/topi/nn/conv2d.py @@ -486,7 +486,6 @@ def conv2d_NCHWc_int8( oc_chunk, ic_chunk_group, kernel_height, kernel_width, _, oc_bn, _ = get_const_tuple( kernel.shape ) - num_filter = oc_chunk * oc_bn groups = ic_chunk // ic_chunk_group dilated_kernel_h = (kernel_height - 1) * dilation_h + 1 diff --git a/python/tvm/topi/x86/conv2d_int8.py b/python/tvm/topi/x86/conv2d_int8.py index b0edb02b0804e..048d9468051b1 100644 --- a/python/tvm/topi/x86/conv2d_int8.py +++ b/python/tvm/topi/x86/conv2d_int8.py @@ -120,7 +120,7 @@ def _pack_data(cfg, data, kernel): kernel = te.compute( (oc_chunk, ic_chunk, kh, kw, ic_bn // n_elems, oc_bn, n_elems), lambda occ, icc, k_h, k_w, icbc, ocb, icbb: kernel[ - occ * oc_bn + ocb, icc * ic_bn + icbc * ic_bn // n_elems + icbb, k_h, k_w + occ * oc_bn + ocb, icc * ic_bn + icbc * n_elems + icbb, k_h, k_w ], name="kernel_vec", )