[Topi x86] Missing vectorize for depthwise conv2d. (#5196)

03ff0cd0 · Animesh Jain · GitHub · 14ae3a6e · 03ff0cd0
Unverified Commit 03ff0cd0 authored Mar 31, 2020 by Animesh Jain Committed by GitHub Mar 31, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 additions and 0 deletions

topi/python/topi/x86/depthwise_conv2d.py
+1 -0

No files found.
--- a/topi/python/topi/x86/depthwise_conv2d.py
+++ b/topi/python/topi/x86/depthwise_conv2d.py
@@ -220,6 +220,7 @@ def _schedule_depthwise_conv2d_NCHWc_impl(s, cfg, data_vec, kernel_vec, conv_out
    _, ic_chunk, oh, ow, ic_block = s[C].op.axis
    ow_chunk, ow_block = s[C].split(ow, factor=tile_ow)
    s[C].reorder(ic_chunk, oh, ow_chunk, ow_block, ic_block)
+    s[C].vectorize(ic_block)
    parallel_axis = s[C].fuse(ic_chunk, oh)
    s[C].parallel(parallel_axis)
    s[CC].compute_at(s[C], ow_chunk)