WmmaSelector< src_type_a, src_type_b, dst_type, MPerWmma, NPerWmma > Struct Template Reference

WmmaSelector&lt; src_type_a, src_type_b, dst_type, MPerWmma, NPerWmma &gt; Struct Template Reference#

Composable Kernel: ck::WmmaSelector< src_type_a, src_type_b, dst_type, MPerWmma, NPerWmma > Struct Template Reference
ck::WmmaSelector< src_type_a, src_type_b, dst_type, MPerWmma, NPerWmma > Struct Template Reference

#include <wmma_gemm.hpp>

Public Member Functions

template<>
constexpr auto GetWmma ()
 
template<>
constexpr auto GetWmma ()
 
template<>
constexpr auto GetWmma ()
 
template<>
constexpr auto GetWmma ()
 
template<>
constexpr auto GetWmma ()
 
__host__ constexpr __device__ WmmaSelector ()
 

Static Public Member Functions

template<typename src_type_a_ , typename src_type_b_ , typename dst_type_ , index_t MPerWmma_, index_t NPerWmma_>
static constexpr auto GetWmma ()
 

Static Public Attributes

static constexpr auto selected_wmma
 

Constructor & Destructor Documentation

◆ WmmaSelector()

template<typename src_type_a , typename src_type_b , typename dst_type , index_t MPerWmma, index_t NPerWmma>
__host__ constexpr __device__ ck::WmmaSelector< src_type_a, src_type_b, dst_type, MPerWmma, NPerWmma >::WmmaSelector ( )
inlineconstexpr

Member Function Documentation

◆ GetWmma() [1/6]

template<typename src_type_a , typename src_type_b , typename dst_type , index_t MPerWmma, index_t NPerWmma>
template<typename src_type_a_ , typename src_type_b_ , typename dst_type_ , index_t MPerWmma_, index_t NPerWmma_>
static constexpr auto ck::WmmaSelector< src_type_a, src_type_b, dst_type, MPerWmma, NPerWmma >::GetWmma ( )
staticconstexpr

◆ GetWmma() [2/6]

template<typename src_type_a , typename src_type_b , typename dst_type , index_t MPerWmma, index_t NPerWmma>
template<>
constexpr auto ck::WmmaSelector< src_type_a, src_type_b, dst_type, MPerWmma, NPerWmma >::GetWmma ( )
inlineconstexpr

◆ GetWmma() [3/6]

template<typename src_type_a , typename src_type_b , typename dst_type , index_t MPerWmma, index_t NPerWmma>
template<>
constexpr auto ck::WmmaSelector< src_type_a, src_type_b, dst_type, MPerWmma, NPerWmma >::GetWmma ( )
inlineconstexpr

◆ GetWmma() [4/6]

template<typename src_type_a , typename src_type_b , typename dst_type , index_t MPerWmma, index_t NPerWmma>
template<>
constexpr auto ck::WmmaSelector< src_type_a, src_type_b, dst_type, MPerWmma, NPerWmma >::GetWmma ( )
inlineconstexpr

◆ GetWmma() [5/6]

template<typename src_type_a , typename src_type_b , typename dst_type , index_t MPerWmma, index_t NPerWmma>
template<>
constexpr auto ck::WmmaSelector< src_type_a, src_type_b, dst_type, MPerWmma, NPerWmma >::GetWmma ( )
inlineconstexpr

◆ GetWmma() [6/6]

template<typename src_type_a , typename src_type_b , typename dst_type , index_t MPerWmma, index_t NPerWmma>
template<>
constexpr auto ck::WmmaSelector< src_type_a, src_type_b, dst_type, MPerWmma, NPerWmma >::GetWmma ( )
inlineconstexpr

Member Data Documentation

◆ selected_wmma

template<typename src_type_a , typename src_type_b , typename dst_type , index_t MPerWmma, index_t NPerWmma>
constexpr auto ck::WmmaSelector< src_type_a, src_type_b, dst_type, MPerWmma, NPerWmma >::selected_wmma
staticconstexpr
Initial value:
=
wmma_type<GetWmma<src_type_a, src_type_b, dst_type, MPerWmma, NPerWmma>(), Number<32>{}>{}

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/tensor_operation/gpu/warp/wmma_gemm.hpp