Update instance selector

27e6da87 · Rostyslav Geyyer · 04240ec7 · 27e6da87
Commit 27e6da87 authored Oct 20, 2023 by Rostyslav Geyyer
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

library/include/ck/library/tensor_operation_instance/gpu/grouped_convolution_backward_weight.hpp ...tion_instance/gpu/grouped_convolution_backward_weight.hpp +2 -1

No files found.
--- a/library/include/ck/library/tensor_operation_instance/gpu/grouped_convolution_backward_weight.hpp
+++ b/library/include/ck/library/tensor_operation_instance/gpu/grouped_convolution_backward_weight.hpp
@@ -780,7 +780,8 @@ struct DeviceOperationInstanceFactory<ck::tensor_operation::device::DeviceGroupe
 #endif
 #ifdef CK_ENABLE_FP16
                if constexpr(is_same_v<InDataType, half_t> && is_same_v<WeiDataType, half_t> &&
-                             is_same_v<OutDataType, half_t>)
+                             is_same_v<OutDataType, half_t> && is_same_v<ComputeTypeA, half_t> &&
+                             is_same_v<ComputeTypeB, half_t>)
                {
 #ifdef DL_KERNELS
                    add_device_grouped_conv3d_bwd_weight_dl_gndhwc_gkzyxc_gndhwk_f16_instances(