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 () |
__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>
|
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_>
|
staticconstexpr |
◆ GetWmma() [2/6]
template<typename src_type_a , typename src_type_b , typename dst_type , index_t MPerWmma, index_t NPerWmma>
template<>
|
inlineconstexpr |
◆ GetWmma() [3/6]
template<typename src_type_a , typename src_type_b , typename dst_type , index_t MPerWmma, index_t NPerWmma>
template<>
|
inlineconstexpr |
◆ GetWmma() [4/6]
template<typename src_type_a , typename src_type_b , typename dst_type , index_t MPerWmma, index_t NPerWmma>
template<>
|
inlineconstexpr |
◆ GetWmma() [5/6]
template<typename src_type_a , typename src_type_b , typename dst_type , index_t MPerWmma, index_t NPerWmma>
template<>
|
inlineconstexpr |
◆ GetWmma() [6/6]
template<typename src_type_a , typename src_type_b , typename dst_type , index_t MPerWmma, index_t NPerWmma>
template<>
|
inlineconstexpr |
Member Data Documentation
◆ selected_wmma
template<typename src_type_a , typename src_type_b , typename dst_type , index_t MPerWmma, index_t NPerWmma>
|
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