gridwise_put_element_1d.hpp File Reference

• include
• ck
• tensor_operation
• gpu
• grid

Classes | Namespaces | Functions

gridwise_put_element_1d.hpp File Reference

#include "ck/utility/data_type.hpp"
#include "ck/tensor_operation/gpu/thread/threadwise_tensor_slice_transfer.hpp"

Go to the source code of this file.

Classes
struct	ck::GridwisePutElement_1D< InGrid1dDesc, InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize >

Namespaces
namespace	ck

Functions
template<typename GridwisePutElementwise1dFunctor, typename InGrid1dDesc, typename InDataType, typename IndexDataType, typename OutDataType, typename ElementwiseOperation>
__global__ void	ck::kernel_put_element_1d (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)