/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/tensor_operation/gpu/device/impl/device_avgpool2d_bwd_nhwc_nhwc.hpp Source File#
device_avgpool2d_bwd_nhwc_nhwc.hpp
Go to the documentation of this file.
float launch_and_time_kernel(const StreamConfig &stream_config, F kernel, dim3 grid_dim, dim3 block_dim, std::size_t lds_byte, Args... args)
Definition: kernel_launch.hpp:13
__host__ constexpr __device__ auto integer_least_multiple(X x, Y y)
Definition: math.hpp:78
__host__ constexpr __device__ auto integer_divide_ceil(X x, Y y)
Definition: math.hpp:72
__host__ constexpr __device__ auto integer_divide_floor(X x, Y y)
Definition: math.hpp:66
__host__ constexpr __device__ index_t gcd(index_t x, index_t y)
Definition: math.hpp:154
Definition: ck.hpp:264
__host__ constexpr __device__ auto make_naive_tensor_descriptor(const Tuple< Lengths... > &lengths, const Tuple< Strides... > &strides)
Definition: tensor_descriptor_helper.hpp:49
__host__ constexpr __device__ auto make_merge_transform(const LowLengths &low_lengths)
Definition: multi_index_transform_helper.hpp:55
__global__ void kernel_reduce_threadwise(const InGridDesc_M_K in_grid_desc_m_k, const OutGridDesc_M out_grid_desc_m, const InElementwiseOperation in_elementwise_op, const AccElementwiseOperation acc_elementwise_op, AccDataType alpha, const InDataType *const __restrict__ p_in_value_global, const IndexDataType *const __restrict__ p_in_index_global, AccDataType beta, OutDataType *const __restrict__ p_out_value_global, IndexDataType *const __restrict__ p_out_index_global)
Definition: gridwise_2d_reduction_threadwise.hpp:28
__host__ constexpr __device__ auto make_freeze_transform(const LowerIndex &low_idx)
Definition: multi_index_transform_helper.hpp:98
__host__ constexpr __device__ auto make_embed_transform(const UpLengths &up_lengths, const Coefficients &coefficients)
Definition: multi_index_transform_helper.hpp:48
__host__ constexpr __device__ auto make_slice_transform(const LowLength &low_length, const SliceBegin &slice_begin, const SliceEnd &slice_end)
Definition: multi_index_transform_helper.hpp:110
__host__ constexpr __device__ auto make_pass_through_transform(const LowLength &low_length)
Definition: multi_index_transform_helper.hpp:12
__host__ constexpr __device__ auto transform_tensor_descriptor(const OldTensorDescriptor &old_tensor_desc, const NewTransforms &new_transforms, NewLowerDimensionOldVisibleIdss, NewUpperDimensionNewVisibleIdss)
Definition: tensor_descriptor.hpp:319
__host__ constexpr __device__ auto make_pad_transform(const LowLength &low_length, const LeftPad &left_pad, const RightPad &right_pad, integral_constant< bool, SkipIsValidCheck >=integral_constant< bool, false >{})
Definition: multi_index_transform_helper.hpp:19
__host__ constexpr __device__ auto make_right_pad_transform(const LowLength &low_length, const RightPadLength &right_pad, integral_constant< bool, SkipIsValidCheck >=integral_constant< bool, false >{})
Definition: multi_index_transform_helper.hpp:37
Definition: stream_config.hpp:10
Definition: gridwise_2d_reduction_threadwise.hpp:84
Definition: sequence.hpp:43
Definition: integral_constant.hpp:10
Definition: reduction_operator.hpp:37
Definition: device_base.hpp:50
Definition: device_base.hpp:61
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:295
int num_reduce_
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:363
const DOutDataType * p_dout_grid_
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:356
std::vector< ck::index_t > dout_n_c_wos_strides_
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:360
std::vector< ck::index_t > dout_n_c_wos_lengths_
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:358
Argument(const DOutDataType *p_dout, DInDataType *p_din, std::vector< ck::index_t > dout_n_c_wos_lengths, std::vector< ck::index_t > din_n_c_wos_length, std::vector< ck::index_t > dout_n_c_wos_strides, std::vector< ck::index_t > din_n_c_wos_strides, std::vector< ck::index_t > window_lengths, std::vector< ck::index_t > window_strides, std::vector< ck::index_t > window_dilations, std::vector< ck::index_t > input_left_pads, std::vector< ck::index_t > input_right_pads)
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:296
DInDataType * p_din_grid_
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:357
std::vector< DinGridDesc_M > din_grid_desc_m_container_
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:365
std::vector< DoutGridDesc_M_K > dout_grid_desc_m_k_container_
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:364
std::vector< ck::index_t > din_n_c_wos_strides_
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:361
Div div_element_op_
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:367
std::vector< ck::index_t > din_n_c_wos_length_
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:359
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:371
float Run(const Argument &arg, const StreamConfig &stream_config=StreamConfig{})
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:372
float Run(const BaseArgument *p_arg, const StreamConfig &stream_config=StreamConfig{}) override
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:414
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:41
remove_cvref_t< tuple_element_t< 0, DoutDinGridDesc > > DoutGridDesc_M_K
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:265
decltype(Make2DGridDescriptor_Out_M_K_In_M({0, 0, 0, 0}, {0, 0, 0, 0}, {0, 0, 0, 0}, {0, 0, 0, 0}, {0, 0}, {0, 0}, {0, 0}, {0, 0}, {0, 0}, {0, 0})) DoutDinGridDesc
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:263
tensor_operation::element_wise::UnaryDivide Div
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:274
static auto Make2DGridDescriptor_Out_M_K_In_M(const std::vector< ck::index_t > &dout_n_c_wos_lengths, const std::vector< ck::index_t > &din_n_c_wos_length, const std::vector< ck::index_t > &dout_n_c_wos_strides, const std::vector< ck::index_t > &din_n_c_wos_strides, const std::vector< ck::index_t > &window_lengths, const std::vector< ck::index_t > &window_strides, const std::vector< ck::index_t > &window_dilations, const std::vector< ck::index_t > &input_left_pads, const std::vector< ck::index_t > &input_right_pads, const std::vector< ck::index_t > &tildes)
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:52
static constexpr auto I1
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:46
GridwiseReduction_mk_to_m_threadwise< DOutDataType, DInDataType, ComputeDataType, int, DoutGridDesc_M_K, DinGridDesc_M, reduce::Add, PassThrough, Div, InMemoryDataOperationEnum::Set, false, BlockSize, MThreadSliceSize, KThreadSliceSize, OutSrcInDstVectorDim, InSrcOutDstVectorSize, InSrcOutDstVectorSize > gridwise_reduce
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:292
tensor_operation::element_wise::PassThrough PassThrough
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:273
static constexpr ck::index_t NDimSpatial
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:43
std::unique_ptr< BaseInvoker > MakeInvokerPointer() override
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:501
static constexpr ck::index_t M_BlockTileSize
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:48
static constexpr index_t OutSrcInDstVectorDim
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:271
static constexpr ck::index_t K_BlockTileSize
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:49
std::string GetTypeString() const override
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:506
std::unique_ptr< BaseArgument > MakeArgumentPointer(const void *p_dout, void *p_din, std::vector< ck::index_t > dout_n_c_wos_lengths, std::vector< ck::index_t > din_n_c_wos_length, std::vector< ck::index_t > dout_n_c_wos_strides, std::vector< ck::index_t > din_n_c_wos_strides, std::vector< ck::index_t > window_lengths, std::vector< ck::index_t > window_strides, std::vector< ck::index_t > window_dilations, std::vector< ck::index_t > input_left_pads, std::vector< ck::index_t > input_right_pads) override
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:459
static constexpr auto I0
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:45
remove_cvref_t< tuple_element_t< 1, DoutDinGridDesc > > DinGridDesc_M
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:266
static bool IsSupportedArgument(const Argument &arg)
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:421
bool IsSupportedArgument(const BaseArgument *p_arg) override
Definition: device_avgpool2d_bwd_nhwc_nhwc.hpp:453
Definition: device_avgpool_bwd.hpp:20
Definition: unary_element_wise_operation.hpp:241
Definition: unary_element_wise_operation.hpp:569