GridwisePutElement_1D< InGrid1dDesc, InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize > Struct Template Reference

GridwisePutElement_1D< InGrid1dDesc, InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize > Struct Template Reference#

Composable Kernel: ck::GridwisePutElement_1D< InGrid1dDesc, InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize > Struct Template Reference

#include <gridwise_put_element_1d.hpp>

Static Public Member Functions
static __device__ void	Run (const InGrid1dDesc &in_grid_1d_desc, const InDataType __restrict__ p_in_global, const IndexDataType __restrict__ p_indices_global, OutDataType *__restrict__ p_out_global, const ElementwiseOperation &elementwise_op)

Static Public Attributes
static constexpr auto	I0 = Number<0>{}
static constexpr auto	thread_buffer_desc_m

Member Function Documentation

◆ Run()

template<typename InGrid1dDesc, typename InDataType, typename IndexDataType, typename OutDataType, typename ElementwiseOperation, InMemoryDataOperationEnum MemOp, index_t InVectorSize>

__device__ void ck::GridwisePutElement_1D< InGrid1dDesc, InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize >::Run	(	const InGrid1dDesc &	in_grid_1d_desc,
		const InDataType *__restrict__	p_in_global,
		const IndexDataType *__restrict__	p_indices_global,
		OutDataType *__restrict__	p_out_global,
		const ElementwiseOperation &	elementwise_op )

inlinestatic

Member Data Documentation

◆ I0

template<typename InGrid1dDesc, typename InDataType, typename IndexDataType, typename OutDataType, typename ElementwiseOperation, InMemoryDataOperationEnum MemOp, index_t InVectorSize>

auto ck::GridwisePutElement_1D< InGrid1dDesc, InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize >::I0 = Number<0>{}

staticconstexpr

◆ thread_buffer_desc_m

template<typename InGrid1dDesc, typename InDataType, typename IndexDataType, typename OutDataType, typename ElementwiseOperation, InMemoryDataOperationEnum MemOp, index_t InVectorSize>

auto ck::GridwisePutElement_1D< InGrid1dDesc, InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize >::thread_buffer_desc_m

staticconstexpr

Initial value:

=

make_naive_tensor_descriptor_packed(make_tuple(Number<InVectorSize>{}))

ck::Number

integral_constant< index_t, N > Number

Definition number.hpp:12

ck::make_naive_tensor_descriptor_packed

__host__ __device__ constexpr auto make_naive_tensor_descriptor_packed(const Tuple< Lengths... > &lengths)

Definition tensor_descriptor_helper.hpp:101

ck::make_tuple

__host__ __device__ constexpr auto make_tuple(Xs &&... xs)