ElementWiseKernel< Problem_, Policy_ > Struct Template Reference#
Public Types |
Public Member Functions |
Static Public Member Functions |
Static Public Attributes |
List of all members
ck_tile::ElementWiseKernel< Problem_, Policy_ > Struct Template Reference
#include <elementwise_kernel.hpp>
Public Types | |
using | Problem = ck_tile::remove_cvref_t< Problem_ > |
using | Policy = ck_tile::remove_cvref_t< Policy_ > |
using | XDataType = ck_tile::remove_cvref_t< typename Problem::XDataType > |
using | ComputeDataType = ck_tile::remove_cvref_t< typename Problem::ComputeDataType > |
using | YDataType = ck_tile::remove_cvref_t< typename Problem::YDataType > |
using | ElementWiseOperation = ck_tile::remove_cvref_t< typename Problem::ElementWiseOperation > |
Public Member Functions | |
template<typename... XDataType, typename Dims > | |
CK_TILE_DEVICE void | operator() (Dims lens, Dims input_strides, Dims output_strides, const tuple< XDataType... > &input_tensors, YDataType *p_y) const |
Static Public Member Functions | |
template<typename... Ints> | |
static CK_TILE_HOST bool | IsSupportedArgument (const ck_tile::tuple< Ints... > &input_sizes) |
Static Public Attributes | |
static constexpr index_t | kBlockSize = Problem::BlockShape::kBlockSize |
Member Typedef Documentation
◆ ComputeDataType
template<typename Problem_ , typename Policy_ >
using ck_tile::ElementWiseKernel< Problem_, Policy_ >::ComputeDataType = ck_tile::remove_cvref_t<typename Problem::ComputeDataType> |
◆ ElementWiseOperation
template<typename Problem_ , typename Policy_ >
using ck_tile::ElementWiseKernel< Problem_, Policy_ >::ElementWiseOperation = ck_tile::remove_cvref_t<typename Problem::ElementWiseOperation> |
◆ Policy
template<typename Problem_ , typename Policy_ >
using ck_tile::ElementWiseKernel< Problem_, Policy_ >::Policy = ck_tile::remove_cvref_t<Policy_> |
◆ Problem
template<typename Problem_ , typename Policy_ >
using ck_tile::ElementWiseKernel< Problem_, Policy_ >::Problem = ck_tile::remove_cvref_t<Problem_> |
◆ XDataType
template<typename Problem_ , typename Policy_ >
using ck_tile::ElementWiseKernel< Problem_, Policy_ >::XDataType = ck_tile::remove_cvref_t<typename Problem::XDataType> |
◆ YDataType
template<typename Problem_ , typename Policy_ >
using ck_tile::ElementWiseKernel< Problem_, Policy_ >::YDataType = ck_tile::remove_cvref_t<typename Problem::YDataType> |
Member Function Documentation
◆ IsSupportedArgument()
template<typename Problem_ , typename Policy_ >
template<typename... Ints>
|
inlinestatic |
◆ operator()()
template<typename Problem_ , typename Policy_ >
template<typename... XDataType, typename Dims >
|
inline |
Member Data Documentation
◆ kBlockSize
template<typename Problem_ , typename Policy_ >
|
staticconstexpr |
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_tile/ops/elementwise/kernel/elementwise_kernel.hpp