GridwiseElementwise_1D< InGrid1dDescTuple, OutGrid1dDescTuple, InDataTypePointerTuple, OutDataTypePointerTuple, ElementwiseOperation, UnaryOperation, Scale, MPerThread, InScalarPerVectorSeq, OutScalarPerVectorSeq > Struct Template Reference

GridwiseElementwise_1D&lt; InGrid1dDescTuple, OutGrid1dDescTuple, InDataTypePointerTuple, OutDataTypePointerTuple, ElementwiseOperation, UnaryOperation, Scale, MPerThread, InScalarPerVectorSeq, OutScalarPerVectorSeq &gt; Struct Template Reference#

Composable Kernel: ck::GridwiseElementwise_1D< InGrid1dDescTuple, OutGrid1dDescTuple, InDataTypePointerTuple, OutDataTypePointerTuple, ElementwiseOperation, UnaryOperation, Scale, MPerThread, InScalarPerVectorSeq, OutScalarPerVectorSeq > Struct Template Reference
Public Types | Static Public Member Functions | Static Public Attributes | List of all members
ck::GridwiseElementwise_1D< InGrid1dDescTuple, OutGrid1dDescTuple, InDataTypePointerTuple, OutDataTypePointerTuple, ElementwiseOperation, UnaryOperation, Scale, MPerThread, InScalarPerVectorSeq, OutScalarPerVectorSeq > Struct Template Reference

#include <gridwise_elementwise_1d_scale.hpp>

Public Types

using PassThroughOp = tensor_operation::element_wise::PassThrough
 

Static Public Member Functions

static __device__ void Run (const InGrid1dDescTuple in_grid_1d_desc_tuple, const OutGrid1dDescTuple out_grid_1d_desc_tuple, const InDataTypePointerTuple p_in_global_tuple, const OutDataTypePointerTuple p_out_global_tuple, const ElementwiseOperation elementwise_op, const UnaryOperation unary_op, const Scale scale_op)
 

Static Public Attributes

static constexpr index_t NumInput = InDataTypePointerTuple::Size()
 
static constexpr index_t NumOutput = OutDataTypePointerTuple::Size()
 
static constexpr auto I0 = Number<0>{}
 
static constexpr auto thread_buffer_desc_m
 

Member Typedef Documentation

◆ PassThroughOp

template<typename InGrid1dDescTuple , typename OutGrid1dDescTuple , typename InDataTypePointerTuple , typename OutDataTypePointerTuple , typename ElementwiseOperation , typename UnaryOperation , typename Scale , index_t MPerThread, typename InScalarPerVectorSeq , typename OutScalarPerVectorSeq >
using ck::GridwiseElementwise_1D< InGrid1dDescTuple, OutGrid1dDescTuple, InDataTypePointerTuple, OutDataTypePointerTuple, ElementwiseOperation, UnaryOperation, Scale, MPerThread, InScalarPerVectorSeq, OutScalarPerVectorSeq >::PassThroughOp = tensor_operation::element_wise::PassThrough

Member Function Documentation

◆ Run()

template<typename InGrid1dDescTuple , typename OutGrid1dDescTuple , typename InDataTypePointerTuple , typename OutDataTypePointerTuple , typename ElementwiseOperation , typename UnaryOperation , typename Scale , index_t MPerThread, typename InScalarPerVectorSeq , typename OutScalarPerVectorSeq >
static __device__ void ck::GridwiseElementwise_1D< InGrid1dDescTuple, OutGrid1dDescTuple, InDataTypePointerTuple, OutDataTypePointerTuple, ElementwiseOperation, UnaryOperation, Scale, MPerThread, InScalarPerVectorSeq, OutScalarPerVectorSeq >::Run ( const InGrid1dDescTuple  in_grid_1d_desc_tuple,
const OutGrid1dDescTuple  out_grid_1d_desc_tuple,
const InDataTypePointerTuple  p_in_global_tuple,
const OutDataTypePointerTuple  p_out_global_tuple,
const ElementwiseOperation  elementwise_op,
const UnaryOperation  unary_op,
const Scale  scale_op 
)
inlinestatic

Member Data Documentation

◆ I0

template<typename InGrid1dDescTuple , typename OutGrid1dDescTuple , typename InDataTypePointerTuple , typename OutDataTypePointerTuple , typename ElementwiseOperation , typename UnaryOperation , typename Scale , index_t MPerThread, typename InScalarPerVectorSeq , typename OutScalarPerVectorSeq >
constexpr auto ck::GridwiseElementwise_1D< InGrid1dDescTuple, OutGrid1dDescTuple, InDataTypePointerTuple, OutDataTypePointerTuple, ElementwiseOperation, UnaryOperation, Scale, MPerThread, InScalarPerVectorSeq, OutScalarPerVectorSeq >::I0 = Number<0>{}
staticconstexpr

◆ NumInput

template<typename InGrid1dDescTuple , typename OutGrid1dDescTuple , typename InDataTypePointerTuple , typename OutDataTypePointerTuple , typename ElementwiseOperation , typename UnaryOperation , typename Scale , index_t MPerThread, typename InScalarPerVectorSeq , typename OutScalarPerVectorSeq >
constexpr index_t ck::GridwiseElementwise_1D< InGrid1dDescTuple, OutGrid1dDescTuple, InDataTypePointerTuple, OutDataTypePointerTuple, ElementwiseOperation, UnaryOperation, Scale, MPerThread, InScalarPerVectorSeq, OutScalarPerVectorSeq >::NumInput = InDataTypePointerTuple::Size()
staticconstexpr

◆ NumOutput

template<typename InGrid1dDescTuple , typename OutGrid1dDescTuple , typename InDataTypePointerTuple , typename OutDataTypePointerTuple , typename ElementwiseOperation , typename UnaryOperation , typename Scale , index_t MPerThread, typename InScalarPerVectorSeq , typename OutScalarPerVectorSeq >
constexpr index_t ck::GridwiseElementwise_1D< InGrid1dDescTuple, OutGrid1dDescTuple, InDataTypePointerTuple, OutDataTypePointerTuple, ElementwiseOperation, UnaryOperation, Scale, MPerThread, InScalarPerVectorSeq, OutScalarPerVectorSeq >::NumOutput = OutDataTypePointerTuple::Size()
staticconstexpr

◆ thread_buffer_desc_m

template<typename InGrid1dDescTuple , typename OutGrid1dDescTuple , typename InDataTypePointerTuple , typename OutDataTypePointerTuple , typename ElementwiseOperation , typename UnaryOperation , typename Scale , index_t MPerThread, typename InScalarPerVectorSeq , typename OutScalarPerVectorSeq >
constexpr auto ck::GridwiseElementwise_1D< InGrid1dDescTuple, OutGrid1dDescTuple, InDataTypePointerTuple, OutDataTypePointerTuple, ElementwiseOperation, UnaryOperation, Scale, MPerThread, InScalarPerVectorSeq, OutScalarPerVectorSeq >::thread_buffer_desc_m
staticconstexpr
Initial value:
=
make_naive_tensor_descriptor_packed(make_tuple(Number<MPerThread>{}))
ck::make_naive_tensor_descriptor_packed
__host__ constexpr __device__ auto make_naive_tensor_descriptor_packed(const Tuple< Lengths... > &lengths)
Definition: tensor_descriptor_helper.hpp:101
ck::make_tuple
__host__ constexpr __device__ auto make_tuple(Xs &&... xs)
Definition: tuple.hpp:211
The documentation for this struct was generated from the following file:
  • /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/tensor_operation/gpu/grid/gridwise_elementwise_1d_scale.hpp