From 7ffa682bd390a8fe298d8526bd52ff7435c6f37a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ville=20Pietil=C3=A4?= <> Date: Fri, 30 Jan 2026 05:05:38 -0500 Subject: [PATCH] Add missing applicability check to v3 fwd convs. --- ...ped_conv_fwd_multiple_abd_xdl_cshuffle_v3.hpp | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/include/ck/tensor_operation/gpu/device/impl/device_grouped_conv_fwd_multiple_abd_xdl_cshuffle_v3.hpp b/include/ck/tensor_operation/gpu/device/impl/device_grouped_conv_fwd_multiple_abd_xdl_cshuffle_v3.hpp index faf8118711..d73006333d 100644 --- a/include/ck/tensor_operation/gpu/device/impl/device_grouped_conv_fwd_multiple_abd_xdl_cshuffle_v3.hpp +++ b/include/ck/tensor_operation/gpu/device/impl/device_grouped_conv_fwd_multiple_abd_xdl_cshuffle_v3.hpp @@ -1614,6 +1614,22 @@ struct DeviceGroupedConvFwdMultipleABD_Xdl_CShuffle_V3 } } } + else if constexpr(ConvForwardSpecialization == ConvolutionForwardSpecialization::Filter3x3) + { + if(C != 1) + { + return false; + } + for(index_t i = 0; i < NDimSpatial; ++i) + { + const index_t filter_spatial_dim = arg.b_g_k_c_xs_lengths_[i + I3]; + + if(filter_spatial_dim != I3) + { + return false; + } + } + } // check vector access of A // FIXME: layout