Invoker Struct Reference

Invoker Struct Reference#

Composable Kernel: ck::tensor_operation::device::DeviceReduceMultiBlock< InDataType, AccDataType, OutDataType, Rank, NumReduceDim, ReduceOperation, InElementwiseOperation, AccElementwiseOperation, OutMemoryDataOperation, PropagateNan, OutputIndex, HaveIndexInputIfOutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcVectorDim, InSrcVectorSize, OutDstVectorSize >::Invoker Struct Reference
Public Member Functions | List of all members
ck::tensor_operation::device::DeviceReduceMultiBlock< InDataType, AccDataType, OutDataType, Rank, NumReduceDim, ReduceOperation, InElementwiseOperation, AccElementwiseOperation, OutMemoryDataOperation, PropagateNan, OutputIndex, HaveIndexInputIfOutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcVectorDim, InSrcVectorSize, OutDstVectorSize >::Invoker Struct Reference

#include <device_reduce_multiblock.hpp>

Inheritance diagram for ck::tensor_operation::device::DeviceReduceMultiBlock< InDataType, AccDataType, OutDataType, Rank, NumReduceDim, ReduceOperation, InElementwiseOperation, AccElementwiseOperation, OutMemoryDataOperation, PropagateNan, OutputIndex, HaveIndexInputIfOutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcVectorDim, InSrcVectorSize, OutDstVectorSize >::Invoker:
ck::tensor_operation::device::BaseInvoker

Public Member Functions

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

Member Function Documentation

◆ Run() [1/2]

template<typename InDataType , typename AccDataType , typename OutDataType , index_t Rank, index_t NumReduceDim, typename ReduceOperation , typename InElementwiseOperation , typename AccElementwiseOperation , InMemoryDataOperationEnum OutMemoryDataOperation, bool PropagateNan, bool OutputIndex, bool HaveIndexInputIfOutputIndex, index_t BlockSize, index_t MThreadClusterSize, index_t KThreadClusterSize, index_t MThreadSliceSize, index_t KThreadSliceSize, index_t InSrcVectorDim, index_t InSrcVectorSize, index_t OutDstVectorSize>
float ck::tensor_operation::device::DeviceReduceMultiBlock< InDataType, AccDataType, OutDataType, Rank, NumReduceDim, ReduceOperation, InElementwiseOperation, AccElementwiseOperation, OutMemoryDataOperation, PropagateNan, OutputIndex, HaveIndexInputIfOutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcVectorDim, InSrcVectorSize, OutDstVectorSize >::Invoker::Run ( const Argument &  arg,
const StreamConfig &  stream_config = StreamConfig{} 
)
inline

◆ Run() [2/2]

template<typename InDataType , typename AccDataType , typename OutDataType , index_t Rank, index_t NumReduceDim, typename ReduceOperation , typename InElementwiseOperation , typename AccElementwiseOperation , InMemoryDataOperationEnum OutMemoryDataOperation, bool PropagateNan, bool OutputIndex, bool HaveIndexInputIfOutputIndex, index_t BlockSize, index_t MThreadClusterSize, index_t KThreadClusterSize, index_t MThreadSliceSize, index_t KThreadSliceSize, index_t InSrcVectorDim, index_t InSrcVectorSize, index_t OutDstVectorSize>
float ck::tensor_operation::device::DeviceReduceMultiBlock< InDataType, AccDataType, OutDataType, Rank, NumReduceDim, ReduceOperation, InElementwiseOperation, AccElementwiseOperation, OutMemoryDataOperation, PropagateNan, OutputIndex, HaveIndexInputIfOutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcVectorDim, InSrcVectorSize, OutDstVectorSize >::Invoker::Run ( const BaseArgument *  p_arg,
const StreamConfig &  stream_config = StreamConfig{} 
)
inlineoverridevirtual

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

The documentation for this struct was generated from the following file:
  • /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-7.0.1/include/ck/tensor_operation/gpu/device/impl/device_reduce_multiblock.hpp