rmsnorm2d_fwd_traits.hpp File Reference

rmsnorm2d_fwd_traits.hpp File Reference#

Composable Kernel: rmsnorm2d_fwd_traits.hpp File Reference
Classes | Namespaces | Enumerations
rmsnorm2d_fwd_traits.hpp File Reference
#include "ck_tile/core/utility/type_traits.hpp"

Go to the source code of this file.

Classes

struct  ck_tile::Rmsnorm2dFusedAddEnumName< Rmsnorm2dFusedAddEnum::NO_ADD >
struct  ck_tile::Rmsnorm2dFusedAddEnumName< Rmsnorm2dFusedAddEnum::PRE_ADD_STORE >
struct  ck_tile::Rmsnorm2dFusedAddEnumName< Rmsnorm2dFusedAddEnum::PRE_ADD >
struct  ck_tile::Rmsnorm2dFusedQuantEnumName< Rmsnorm2dFusedQuantEnum::NO_SWEEP >
struct  ck_tile::Rmsnorm2dFusedQuantEnumName< Rmsnorm2dFusedQuantEnum::DYNAMIC_QUANT >
struct  ck_tile::Rmsnorm2dFusedQuantEnumName< Rmsnorm2dFusedQuantEnum::SMOOTH_DYNAMIC_QUANT >
struct  ck_tile::Rmsnorm2dSensitiveEnumName< Rmsnorm2dSensitiveEnum::NO_SPECIFIC_MODEL >
struct  ck_tile::Rmsnorm2dSensitiveEnumName< Rmsnorm2dSensitiveEnum::T5_MODEL_LIKE >
struct  ck_tile::Rmsnorm2dFwdTraits< kPadN_, kSaveInvRms_, kSaveUnquant_, kTwoPass_, kFusedAdd_, kFusedQuant_, kUseModelSensitiveRMSNorm_ >

Namespaces

namespace  ck_tile

Enumerations

enum class  ck_tile::Rmsnorm2dFusedAddEnum {
  ck_tile::NO_ADD = 0 ,
  ck_tile::PRE_ADD_STORE = 1 ,
  ck_tile::PRE_ADD = 2
}
enum class  ck_tile::Rmsnorm2dFusedQuantEnum {
  ck_tile::NO_SWEEP = 0 ,
  ck_tile::SMOOTH_DYNAMIC_QUANT = 1 ,
  ck_tile::DYNAMIC_QUANT = 2
}
enum class  ck_tile::Rmsnorm2dSensitiveEnum {
  ck_tile::NO_SPECIFIC_MODEL = 0 ,
  ck_tile::T5_MODEL_LIKE = 1
}