[GPU][MTL] choose better conv kernel for filter size 32x32 in iGPU case

riverlijunjie · Jun 15, 2024 · 6414481 · 6414481
1 parent 8be4a62
commit 6414481
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/...ntel_gpu/src/kernel_selector/kernels/convolution/convolution_kernel_bfyx_os_iyx_osv16.cpp b/...ntel_gpu/src/kernel_selector/kernels/convolution/convolution_kernel_bfyx_os_iyx_osv16.cpp
@@ -212,7 +212,7 @@ bool ConvolutionKernel_bfyx_os_iyx_osv16::Validate(const Params& p) const {
     const size_t acceptable_filter_size = 1024;     // This acceptable size was decided by heuristics
     const auto& params = static_cast<const convolution_params&>(p);
     auto filter_size = params.filterSize.x * params.filterSize.y;
-    if (filter_size > acceptable_filter_size) {
+    if (filter_size >= acceptable_filter_size) {
         return false;
     }
 

diff --git a/...ntel_gpu/src/kernel_selector/kernels/convolution/convolution_kernel_bfyx_os_iyx_osv32.cpp b/...ntel_gpu/src/kernel_selector/kernels/convolution/convolution_kernel_bfyx_os_iyx_osv32.cpp
@@ -196,7 +196,7 @@ bool ConvolutionKernel_bfyx_os_iyx_osv32::Validate(const Params& p) const {
     const size_t acceptable_filter_size = 1024;     // This acceptable size was decided by heuristics
     const auto& params = static_cast<const convolution_params&>(p);
     auto filter_size = params.filterSize.x * params.filterSize.y;
-    if (filter_size > acceptable_filter_size) {
+    if (filter_size >= acceptable_filter_size) {
         return false;
     }