From 6d9d210d497dce6c2a5274a704a37cae42e5e81a Mon Sep 17 00:00:00 2001 From: Egor Churaev Date: Tue, 31 Aug 2021 11:15:29 +0300 Subject: [PATCH] Update values of vectorization factor --- python/tvm/topi/gpu/conv2d_nhwc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/topi/gpu/conv2d_nhwc.py b/python/tvm/topi/gpu/conv2d_nhwc.py index 0aa66845e8978..a52ed408b90fa 100644 --- a/python/tvm/topi/gpu/conv2d_nhwc.py +++ b/python/tvm/topi/gpu/conv2d_nhwc.py @@ -54,7 +54,7 @@ def schedule_conv2d_nhwc_direct(cfg, s, Conv): cfg.define_knob("vthread_n", [1] if dynamic_batch else [1, 2]) cfg.define_knob("vthread_c", [1, 2]) cfg.define_knob("step", [16, 3, 32, 64]) - cfg.define_knob("vectorize", [1, 2, 4, 8, 16]) + cfg.define_knob("vectorize", [1, 2, 4, 8]) # fallback support target = tvm.target.Target.current()