Update values of vectorization factor

apache · Sep 2, 2021 · 256f524 · 256f524
1 parent 9e96c63
commit 256f524
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()