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
Public Member Functions | Static Public Member Functions | Static Public Attributes | List of all members
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 ()
template<>
constexpr auto GetWmma ()
template<>
constexpr auto GetWmma ()
template<>
constexpr auto GetWmma ()
template<>
constexpr auto GetWmma ()
__host__ __device__ constexpr 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__ __device__ constexpr ck::WmmaSelector< src_type_a, src_type_b, dst_type, MPerWmma, NPerWmma >::WmmaSelector ( )
inlineconstexpr

Member Function Documentation

◆ GetWmma() [1/10]

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

◆ GetWmma() [2/10]

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

◆ GetWmma() [3/10]

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

◆ GetWmma() [4/10]

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

◆ GetWmma() [5/10]

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

◆ GetWmma() [6/10]

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

◆ GetWmma() [7/10]

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

◆ GetWmma() [8/10]

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

◆ GetWmma() [9/10]

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

◆ GetWmma() [10/10]

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_>
constexpr auto ck::WmmaSelector< src_type_a, src_type_b, dst_type, MPerWmma, NPerWmma >::GetWmma ( )
staticconstexpr

Member Data Documentation

◆ selected_wmma

template<typename src_type_a, typename src_type_b, typename dst_type, index_t MPerWmma, index_t NPerWmma>
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>{}>{}
ck::Number
integral_constant< index_t, N > Number
Definition number.hpp:12
ck::wmma_type
Definition wmma_gemm.hpp:84
The documentation for this struct was generated from the following file: