blockwise_gemm_wmma.hpp File Reference

• include
• ck
• tensor_operation
• gpu
• block

Classes | Namespaces | Macros

blockwise_gemm_wmma.hpp File Reference

#include "ck/utility/common_header.hpp"
#include "ck/tensor_operation/gpu/thread/threadwise_tensor_slice_transfer.hpp"
#include "ck/tensor_operation/gpu/warp/wmma_gemm.hpp"
#include "ck/tensor_description/tensor_adaptor.hpp"
#include "ck/tensor_operation/gpu/element/unary_element_wise_operation.hpp"

Go to the source code of this file.

Classes
struct	ck::BlockwiseGemmWMMA< BlockSize, FloatA, FloatB, FloatAcc, ABlockDesc, BBlockDesc, MPerBlock, NPerBlock, KPerBlock, MPerWMMA, NPerWMMA, MRepeat, NRepeat, KPack, AEnableLds, BEnableLds, TransposeC >
struct	ck::BlockwiseGemmWMMA< BlockSize, FloatA, FloatB, FloatAcc, ABlockDesc, BBlockDesc, MPerBlock, NPerBlock, KPerBlock, MPerWMMA, NPerWMMA, MRepeat, NRepeat, KPack, AEnableLds, BEnableLds, TransposeC >::AThreadCopySelector< true >
struct	ck::BlockwiseGemmWMMA< BlockSize, FloatA, FloatB, FloatAcc, ABlockDesc, BBlockDesc, MPerBlock, NPerBlock, KPerBlock, MPerWMMA, NPerWMMA, MRepeat, NRepeat, KPack, AEnableLds, BEnableLds, TransposeC >::AThreadCopySelector< false >
struct	ck::BlockwiseGemmWMMA< BlockSize, FloatA, FloatB, FloatAcc, ABlockDesc, BBlockDesc, MPerBlock, NPerBlock, KPerBlock, MPerWMMA, NPerWMMA, MRepeat, NRepeat, KPack, AEnableLds, BEnableLds, TransposeC >::BThreadCopySelector< true >
struct	ck::BlockwiseGemmWMMA< BlockSize, FloatA, FloatB, FloatAcc, ABlockDesc, BBlockDesc, MPerBlock, NPerBlock, KPerBlock, MPerWMMA, NPerWMMA, MRepeat, NRepeat, KPack, AEnableLds, BEnableLds, TransposeC >::BThreadCopySelector< false >

Namespaces
namespace	ck

Macros
#define	CK_MNK_LOOP

Macro Definition Documentation

◆ CK_MNK_LOOP

#define CK_MNK_LOOP