// SPDX-License-Identifier: MIT
// Copyright (c) 2025, Advanced Micro Devices, Inc. All rights reserved.

#include "ck/library/tensor_operation_instance/add_device_operation_instance.hpp"
#include "ck/library/tensor_operation_instance/gpu/grouped_conv_fwd/device_grouped_conv_fwd_xdl_mem_instance.hpp"
#include "ck/utility/filter_tuple.hpp"

namespace ck {
namespace tensor_operation {
namespace device {
namespace instance {

using device_grouped_conv3d_fwd_bias_bn_clamp_xdl_ndhwgc_gkzyxc_ndhwgk_f32_tf32_mem_inter_instances =
    std::vector<std::unique_ptr<
        DeviceGroupedConvFwdMultipleABD<3,
                                        NDHWGC,
                                        GKZYXC,
                                        Tuple<NDHWGK, NDHWGK, NDHWGK, NDHWGK, NDHWGK>,
                                        NDHWGK,
                                        F32,
                                        F32,
                                        Tuple<F32, F32, F32, F32, F32>,
                                        F32,
                                        PassThrough,
                                        PassThrough,
                                        BiasNormalizeInInferClamp,
                                        TF32,
                                        TF32>>>;

// Compilation parameters for in[n, hi, wi, g, c] * wei[g, k, y, x, c] = out[n, ho, wo, g, k]
template <int Shards, int ShardIndex>
void add_device_grouped_conv3d_fwd_bias_bn_clamp_xdl_ndhwgc_gkzyxc_ndhwgk_f32_tf32_mem_inter_instances_shard(
    device_grouped_conv3d_fwd_bias_bn_clamp_xdl_ndhwgc_gkzyxc_ndhwgk_f32_tf32_mem_inter_instances&
        instances)
{
    add_device_operation_instances(
        instances,
        ck::util::filter_tuple_by_modulo_t<device_grouped_conv_fwd_xdl_f32_tf32_mem_instances<
                                               3,
                                               NDHWGC,
                                               GKZYXC,
                                               Tuple<NDHWGK, NDHWGK, NDHWGK, NDHWGK, NDHWGK>,
                                               NDHWGK,
                                               ConvFwdDefault,
                                               Interwave,
                                               Tuple<F32, F32, F32, F32, F32>,
                                               BiasNormalizeInInferClamp>,
                                           Shards,
                                           ShardIndex>{});

    add_device_operation_instances(
        instances,
        ck::util::filter_tuple_by_modulo_t<device_grouped_conv_fwd_xdl_f32_tf32_mem_instances<
                                               3,
                                               NDHWGC,
                                               GKZYXC,
                                               Tuple<NDHWGK, NDHWGK, NDHWGK, NDHWGK, NDHWGK>,
                                               NDHWGK,
                                               ConvFwd1x1P0,
                                               Interwave,
                                               Tuple<F32, F32, F32, F32, F32>,
                                               BiasNormalizeInInferClamp>,
                                           Shards,
                                           ShardIndex>{});

    add_device_operation_instances(
        instances,
        ck::util::filter_tuple_by_modulo_t<device_grouped_conv_fwd_xdl_f32_tf32_mem_instances<
                                               3,
                                               NDHWGC,
                                               GKZYXC,
                                               Tuple<NDHWGK, NDHWGK, NDHWGK, NDHWGK, NDHWGK>,
                                               NDHWGK,
                                               ConvFwd1x1S1P0,
                                               Interwave,
                                               Tuple<F32, F32, F32, F32, F32>,
                                               BiasNormalizeInInferClamp>,
                                           Shards,
                                           ShardIndex>{});
}

} // namespace instance
} // namespace device
} // namespace tensor_operation
} // namespace ck
