Add missing applicability check to v3 fwd convs.

This commit is contained in:
Ville Pietilä
2026-01-30 05:05:38 -05:00
parent 966706bb21
commit 7ffa682bd3

View File

@@ -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