BlockRotaryEmbedding< RotaryEnum, ComputeDataType > Struct Template Reference

• ck_tile
• BlockRotaryEmbedding

Static Public Member Functions | List of all members

ck_tile::BlockRotaryEmbedding< RotaryEnum, ComputeDataType > Struct Template Reference

#include <block_rotary_embedding.hpp>

Static Public Member Functions
template<typename DistributedTensor , typename OtherDramBlockWindow , typename RotaryCosDramBlockWindow , typename RotarySinDramBlockWindow >
static CK_TILE_HOST_DEVICE void	apply (DistributedTensor &tile, OtherDramBlockWindow other_window, RotaryCosDramBlockWindow rotary_cos_window, RotarySinDramBlockWindow rotary_sin_window, index_t rotary_dim, index_t thread_end)

Member Function Documentation

◆ apply()

template<RotaryEmbeddingEnum RotaryEnum, typename ComputeDataType = float>

template<typename DistributedTensor , typename OtherDramBlockWindow , typename RotaryCosDramBlockWindow , typename RotarySinDramBlockWindow >

static CK_TILE_HOST_DEVICE void ck_tile::BlockRotaryEmbedding< RotaryEnum, ComputeDataType >::apply ( DistributedTensor &  tile, OtherDramBlockWindow  other_window, RotaryCosDramBlockWindow  rotary_cos_window, RotarySinDramBlockWindow  rotary_sin_window, index_t  rotary_dim, index_t  thread_end  )

inlinestatic

---

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-6.4.3/include/ck_tile/ops/fmha/block/block_rotary_embedding.hpp