minor cleanup

apache · Feb 13, 2022 · 35f6bb1 · 35f6bb1
1 parent cbfe979
commit 35f6bb1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/python/tvm/topi/x86/dense.py b/python/tvm/topi/x86/dense.py
@@ -333,18 +333,18 @@ def split_y(out):
 
     if C == O:
         fused = s[O].fuse(a_yo, a_xo)
-        s[O].parallel(fused)
     else:
         a_yo, a_yi = split_y(O)
         a_xo, a_xi = s[O].split(O.op.axis[1], factor=16)
 
         s[O].reorder(a_yo, a_xo, a_yi, a_xi)
-        fused = s[O].fuse(a_yo, a_xo)
         s[O].vectorize(a_xi)
-        s[O].parallel(fused)
-
         s[C].compute_at(s[O], a_yi)
 
+        fused = s[O].fuse(a_yo, a_xo)
+
+    s[O].parallel(fused)
+
     return s