x64: brgemm 1x1 conv: fix maybe_rtus

oneapi-src · Dec 23, 2023 · b5f916e · b5f916e
1 parent ee3b12d
commit b5f916e
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/src/cpu/x64/jit_brgemm_1x1_conv.cpp b/src/cpu/x64/jit_brgemm_1x1_conv.cpp
@@ -282,11 +282,13 @@ void brgemm_1x1_convolution_fwd_t<isa>::maybe_rtus(int ithr,
     while (od < OD) {
         // copy to end of column
         const auto nh = nstl::min(count / OW, OH - oh);
-        call_kernel(nh, 0, od, oh, ow);
-        count -= nh * OW;
-        if (count == 0) return;
-        oh = (oh + nh) % OH;
-        if (oh == 0) od++;
+        if (nh > 0) {
+            call_kernel(nh, 0, od, oh, ow);
+            count -= nh * OW;
+            if (count == 0) return;
+            oh = (oh + nh) % OH;
+            if (oh == 0) od++;
+        }
         if (count < OW) {
             // copy partial row
             const auto nw = count;