WarpGemmAttributeWmmaImpl< Traits > Struct Template Reference

WarpGemmAttributeWmmaImpl&lt; Traits &gt; Struct Template Reference#

Composable Kernel: ck_tile::WarpGemmAttributeWmmaImpl< Traits > Struct Template Reference
Public Types | Public Member Functions | Static Public Attributes | List of all members
ck_tile::WarpGemmAttributeWmmaImpl< Traits > Struct Template Reference

#include <warp_gemm_attribute_wmma_impl.hpp>

Public Types

using ADataType = typename Traits::ADataType
using BDataType = typename Traits::BDataType
using CDataType = typename Traits::CDataType
using AVecType = typename Traits::AVecType
using BVecType = typename Traits::BVecType
using CVecType = typename Traits::CVecType
using kABPs2RHssMajor = typename Traits::kABPs2RHssMajor
using kABPs2RHssMinor = typename Traits::kABPs2RHssMinor
using kABYs2RHsMajor = typename Traits::kABYs2RHsMajor
using kABYs2RHsMinor = typename Traits::kABYs2RHsMinor
using kCPs2RHssMajor = typename Traits::kCPs2RHssMajor
using kCPs2RHssMinor = typename Traits::kCPs2RHssMinor
using kCYs2RHsMajor = typename Traits::kCYs2RHsMajor
using kCYs2RHsMinor = typename Traits::kCYs2RHsMinor
using kCTPs2RHssMajor = typename Traits::kCTPs2RHssMajor
using kCTPs2RHssMinor = typename Traits::kCTPs2RHssMinor
using kCTYs2RHsMajor = typename Traits::kCTYs2RHsMajor
using kCTYs2RHsMinor = typename Traits::kCTYs2RHsMinor

Public Member Functions

template<bool clamp = false, bool post_nop_ = false>
CK_TILE_DEVICE void operator() (CVecType &c_vec, const AVecType &a_vec, const BVecType &b_vec, bool_constant< post_nop_ >={}) const
template<bool clamp = false>
CK_TILE_DEVICE CVecType operator() (const AVecType &a_vec, const BVecType &b_vec) const

Static Public Attributes

static constexpr index_t kM = Traits::kM
static constexpr index_t kN = Traits::kN
static constexpr index_t kK = Traits::kK
static constexpr index_t kAMBlock = Traits::kAMBlock
static constexpr index_t kBNBlock = Traits::kBNBlock
static constexpr index_t kRepeat = Traits::kRepeat
static constexpr index_t kAMLane = Traits::kAMLane
static constexpr index_t kBNLane = Traits::kBNLane
static constexpr index_t kABK0PerLane = Traits::kABK0PerLane
static constexpr index_t kABKLane = Traits::kABKLane
static constexpr index_t kABK1PerLane = Traits::kABK1PerLane
static constexpr index_t kCMLane = Traits::kCMLane
static constexpr index_t kCNLane = Traits::kCNLane
static constexpr index_t kCM0PerLane = Traits::kCM0PerLane
static constexpr index_t kCM1PerLane = Traits::kCM1PerLane

Member Typedef Documentation

◆ ADataType

template<typename Traits>
using ck_tile::WarpGemmAttributeWmmaImpl< Traits >::ADataType = typename Traits::ADataType

◆ AVecType

template<typename Traits>
using ck_tile::WarpGemmAttributeWmmaImpl< Traits >::AVecType = typename Traits::AVecType

◆ BDataType

template<typename Traits>
using ck_tile::WarpGemmAttributeWmmaImpl< Traits >::BDataType = typename Traits::BDataType

◆ BVecType

template<typename Traits>
using ck_tile::WarpGemmAttributeWmmaImpl< Traits >::BVecType = typename Traits::BVecType

◆ CDataType

template<typename Traits>
using ck_tile::WarpGemmAttributeWmmaImpl< Traits >::CDataType = typename Traits::CDataType

◆ CVecType

template<typename Traits>
using ck_tile::WarpGemmAttributeWmmaImpl< Traits >::CVecType = typename Traits::CVecType

◆ kABPs2RHssMajor

template<typename Traits>
using ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kABPs2RHssMajor = typename Traits::kABPs2RHssMajor

◆ kABPs2RHssMinor

template<typename Traits>
using ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kABPs2RHssMinor = typename Traits::kABPs2RHssMinor

◆ kABYs2RHsMajor

template<typename Traits>
using ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kABYs2RHsMajor = typename Traits::kABYs2RHsMajor

◆ kABYs2RHsMinor

template<typename Traits>
using ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kABYs2RHsMinor = typename Traits::kABYs2RHsMinor

◆ kCPs2RHssMajor

template<typename Traits>
using ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kCPs2RHssMajor = typename Traits::kCPs2RHssMajor

◆ kCPs2RHssMinor

template<typename Traits>
using ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kCPs2RHssMinor = typename Traits::kCPs2RHssMinor

◆ kCTPs2RHssMajor

template<typename Traits>
using ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kCTPs2RHssMajor = typename Traits::kCTPs2RHssMajor

◆ kCTPs2RHssMinor

template<typename Traits>
using ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kCTPs2RHssMinor = typename Traits::kCTPs2RHssMinor

◆ kCTYs2RHsMajor

template<typename Traits>
using ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kCTYs2RHsMajor = typename Traits::kCTYs2RHsMajor

◆ kCTYs2RHsMinor

template<typename Traits>
using ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kCTYs2RHsMinor = typename Traits::kCTYs2RHsMinor

◆ kCYs2RHsMajor

template<typename Traits>
using ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kCYs2RHsMajor = typename Traits::kCYs2RHsMajor

◆ kCYs2RHsMinor

template<typename Traits>
using ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kCYs2RHsMinor = typename Traits::kCYs2RHsMinor

Member Function Documentation

◆ operator()() [1/2]

template<typename Traits>
template<bool clamp = false>
CK_TILE_DEVICE CVecType ck_tile::WarpGemmAttributeWmmaImpl< Traits >::operator() ( const AVecType & a_vec,
const BVecType & b_vec ) const
inline

◆ operator()() [2/2]

template<typename Traits>
template<bool clamp = false, bool post_nop_ = false>
CK_TILE_DEVICE void ck_tile::WarpGemmAttributeWmmaImpl< Traits >::operator() ( CVecType & c_vec,
const AVecType & a_vec,
const BVecType & b_vec,
bool_constant< post_nop_ > = {} ) const
inline

Member Data Documentation

◆ kABK0PerLane

template<typename Traits>
index_t ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kABK0PerLane = Traits::kABK0PerLane
staticconstexpr

◆ kABK1PerLane

template<typename Traits>
index_t ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kABK1PerLane = Traits::kABK1PerLane
staticconstexpr

◆ kABKLane

template<typename Traits>
index_t ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kABKLane = Traits::kABKLane
staticconstexpr

◆ kAMBlock

template<typename Traits>
index_t ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kAMBlock = Traits::kAMBlock
staticconstexpr

◆ kAMLane

template<typename Traits>
index_t ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kAMLane = Traits::kAMLane
staticconstexpr

◆ kBNBlock

template<typename Traits>
index_t ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kBNBlock = Traits::kBNBlock
staticconstexpr

◆ kBNLane

template<typename Traits>
index_t ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kBNLane = Traits::kBNLane
staticconstexpr

◆ kCM0PerLane

template<typename Traits>
index_t ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kCM0PerLane = Traits::kCM0PerLane
staticconstexpr

◆ kCM1PerLane

template<typename Traits>
index_t ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kCM1PerLane = Traits::kCM1PerLane
staticconstexpr

◆ kCMLane

template<typename Traits>
index_t ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kCMLane = Traits::kCMLane
staticconstexpr

◆ kCNLane

template<typename Traits>
index_t ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kCNLane = Traits::kCNLane
staticconstexpr

◆ kK

template<typename Traits>
index_t ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kK = Traits::kK
staticconstexpr

◆ kM

template<typename Traits>
index_t ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kM = Traits::kM
staticconstexpr

◆ kN

template<typename Traits>
index_t ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kN = Traits::kN
staticconstexpr

◆ kRepeat

template<typename Traits>
index_t ck_tile::WarpGemmAttributeWmmaImpl< Traits >::kRepeat = Traits::kRepeat
staticconstexpr
The documentation for this struct was generated from the following file: