Invoker Struct Reference

Invoker Struct Reference#

Composable Kernel: ck::tensor_operation::device::DevicePermuteImpl< NumDim, InDataType, OutDataType, ElementwiseOperation, BlockSize, NPerBlock, HPerBlock, WPerBlock, InBlockLdsExtraW, InBlockTransferThreadClusterLengths, InBlockTransferThreadClusterArrangeOrder, SrcVectorDim, DstVectorDim, SrcScalarPerVector, DstScalarPerVector >::Invoker Struct Reference
Public Member Functions | Static Public Member Functions | List of all members
ck::tensor_operation::device::DevicePermuteImpl< NumDim, InDataType, OutDataType, ElementwiseOperation, BlockSize, NPerBlock, HPerBlock, WPerBlock, InBlockLdsExtraW, InBlockTransferThreadClusterLengths, InBlockTransferThreadClusterArrangeOrder, SrcVectorDim, DstVectorDim, SrcScalarPerVector, DstScalarPerVector >::Invoker Struct Reference

#include <device_permute_impl.hpp>

Inheritance diagram for ck::tensor_operation::device::DevicePermuteImpl< NumDim, InDataType, OutDataType, ElementwiseOperation, BlockSize, NPerBlock, HPerBlock, WPerBlock, InBlockLdsExtraW, InBlockTransferThreadClusterLengths, InBlockTransferThreadClusterArrangeOrder, SrcVectorDim, DstVectorDim, SrcScalarPerVector, DstScalarPerVector >::Invoker:
ck::tensor_operation::device::BaseInvoker

Public Member Functions

float Run (const BaseArgument *arg, const StreamConfig &stream_config=StreamConfig{}) override final
Public Member Functions inherited from ck::tensor_operation::device::BaseInvoker
 BaseInvoker ()=default
 BaseInvoker (const BaseInvoker &)=default
BaseInvoker & operator= (const BaseInvoker &)=default
virtual ~BaseInvoker ()

Static Public Member Functions

static float Run (const Argument &arg, const StreamConfig &stream_config=StreamConfig{})

Member Function Documentation

◆ Run() [1/2]

template<index_t NumDim, typename InDataType, typename OutDataType, typename ElementwiseOperation, index_t BlockSize, index_t NPerBlock, index_t HPerBlock, index_t WPerBlock, index_t InBlockLdsExtraW, typename InBlockTransferThreadClusterLengths, typename InBlockTransferThreadClusterArrangeOrder, index_t SrcVectorDim, index_t DstVectorDim, index_t SrcScalarPerVector, index_t DstScalarPerVector>
float ck::tensor_operation::device::DevicePermuteImpl< NumDim, InDataType, OutDataType, ElementwiseOperation, BlockSize, NPerBlock, HPerBlock, WPerBlock, InBlockLdsExtraW, InBlockTransferThreadClusterLengths, InBlockTransferThreadClusterArrangeOrder, SrcVectorDim, DstVectorDim, SrcScalarPerVector, DstScalarPerVector >::Invoker::Run ( const Argument & arg,
const StreamConfig & stream_config = StreamConfig{} )
inlinestatic

◆ Run() [2/2]

template<index_t NumDim, typename InDataType, typename OutDataType, typename ElementwiseOperation, index_t BlockSize, index_t NPerBlock, index_t HPerBlock, index_t WPerBlock, index_t InBlockLdsExtraW, typename InBlockTransferThreadClusterLengths, typename InBlockTransferThreadClusterArrangeOrder, index_t SrcVectorDim, index_t DstVectorDim, index_t SrcScalarPerVector, index_t DstScalarPerVector>
float ck::tensor_operation::device::DevicePermuteImpl< NumDim, InDataType, OutDataType, ElementwiseOperation, BlockSize, NPerBlock, HPerBlock, WPerBlock, InBlockLdsExtraW, InBlockTransferThreadClusterLengths, InBlockTransferThreadClusterArrangeOrder, SrcVectorDim, DstVectorDim, SrcScalarPerVector, DstScalarPerVector >::Invoker::Run ( const BaseArgument * arg,
const StreamConfig & stream_config = StreamConfig{} )
inlinefinaloverridevirtual

Reimplemented from ck::tensor_operation::device::BaseInvoker.

The documentation for this struct was generated from the following file: