Skip to content

Commit 35f6bb1

Browse files
committed
minor cleanup
1 parent cbfe979 commit 35f6bb1

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

python/tvm/topi/x86/dense.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -333,18 +333,18 @@ def split_y(out):
333333

334334
if C == O:
335335
fused = s[O].fuse(a_yo, a_xo)
336-
s[O].parallel(fused)
337336
else:
338337
a_yo, a_yi = split_y(O)
339338
a_xo, a_xi = s[O].split(O.op.axis[1], factor=16)
340339

341340
s[O].reorder(a_yo, a_xo, a_yi, a_xi)
342-
fused = s[O].fuse(a_yo, a_xo)
343341
s[O].vectorize(a_xi)
344-
s[O].parallel(fused)
345-
346342
s[C].compute_at(s[O], a_yi)
347343

344+
fused = s[O].fuse(a_yo, a_xo)
345+
346+
s[O].parallel(fused)
347+
348348
return s
349349

350350

0 commit comments

Comments
 (0)