[HotFix] add two missing cases in #2869 (#2976)

ROCm · May 23, 2024 · 7d58b3f · 7d58b3f
1 parent 89d0dfa
commit 7d58b3f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/solver/conv_asm_implicit_gemm_gtc_wrw_nhwc.cpp b/src/solver/conv_asm_implicit_gemm_gtc_wrw_nhwc.cpp
@@ -877,8 +877,12 @@ bool ConvAsmImplicitGemmGTCDynamicWrwXdlopsNHWC::IsApplicable(
         const auto pad_h    = problem.GetPadH();
         const auto pad_w    = problem.GetPadW();
 
-        if(c == 1 && k == 1 && hi == 1 && wi == 1 && y == 3 && x == 3 && pad_h == 2 && pad_w == 2 &&
-           stride_h == 2 && stride_w == 2)
+        if((c == 1 && k == 1 && hi == 1 && wi == 1 && y == 3 && x == 3 && pad_h == 2 &&
+            pad_w == 2 && stride_h == 2 && stride_w == 2) ||
+           (c == 1 && k == 1 && hi == 1 && wi == 5 && y == 3 && x == 6 && pad_h == 2 &&
+            pad_w == 1 && stride_h == 2 && stride_w == 1) ||
+           (c == 1 && k == 1 && hi == 1 && wi == 1 && y == 2 && x == 3 && pad_h == 1 &&
+            pad_w == 2 && stride_h == 1 && stride_w == 2))
             return false;
     }
 #endif