From 0beacdf340f078c0b158400593fb077d0ea30245 Mon Sep 17 00:00:00 2001
From: Masahiro Masuda <masahi129@gmail.com>
Date: Thu, 24 Mar 2022 11:33:07 +0900
Subject: [PATCH] int32_lanes=4 by default

---
 python/tvm/topi/x86/conv2d_alter_op.py | 1 +
 python/tvm/topi/x86/conv2d_int8.py     | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/python/tvm/topi/x86/conv2d_alter_op.py b/python/tvm/topi/x86/conv2d_alter_op.py
index 9234581f1d5b..032c0e2e236b 100644
--- a/python/tvm/topi/x86/conv2d_alter_op.py
+++ b/python/tvm/topi/x86/conv2d_alter_op.py
@@ -159,6 +159,7 @@ def _alter_conv2d_layout(attrs, inputs, tinfos, out_type):
                 out_dtype,
                 False,
                 data_layout,
+                int32_lanes=16,
             )
 
         batch_size, in_channel, height, width = get_const_tuple(data_tensor.shape)
diff --git a/python/tvm/topi/x86/conv2d_int8.py b/python/tvm/topi/x86/conv2d_int8.py
index 8a20dfa6974c..b0edb02b0804 100644
--- a/python/tvm/topi/x86/conv2d_int8.py
+++ b/python/tvm/topi/x86/conv2d_int8.py
@@ -43,7 +43,7 @@ def _get_default_config_int8(
     out_dtype,
     is_depthwise=False,
     layout="NCHW",
-    int32_lanes=16,
+    int32_lanes=4,
 ):
     """
     Get default schedule config for the workload
@@ -172,6 +172,7 @@ def conv2d_NCHWc_int8(cfg, data, kernel, strides, padding, dilation, layout, out
             padding,
             dilation,
             out_dtype,
+            int32_lanes=16,
         )
 
     # Pack data if raw 4-D data is provided.