/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/rmsnorm2d/pipeline/rmsnorm2d_fwd_traits.hpp Source File#
rmsnorm2d_fwd_traits.hpp
Go to the documentation of this file.
21 template<> struct Rmsnorm2dFusedAddEnumName<Rmsnorm2dFusedAddEnum::NO_ADD> { static constexpr const char * name = "no"; };
22 template<> struct Rmsnorm2dFusedAddEnumName<Rmsnorm2dFusedAddEnum::PRE_ADD_STORE> { static constexpr const char * name = "pras"; };
23 template<> struct Rmsnorm2dFusedAddEnumName<Rmsnorm2dFusedAddEnum::PRE_ADD> { static constexpr const char * name = "pra"; };
35 template<> struct Rmsnorm2dFusedQuantEnumName<Rmsnorm2dFusedQuantEnum::NO_SWEEP> { static constexpr const char * name = "no"; };
36 template<> struct Rmsnorm2dFusedQuantEnumName<Rmsnorm2dFusedQuantEnum::DYNAMIC_QUANT> { static constexpr const char * name = "dqt"; };
37 template<> struct Rmsnorm2dFusedQuantEnumName<Rmsnorm2dFusedQuantEnum::SMOOTH_DYNAMIC_QUANT> { static constexpr const char * name = "smdqt"; };
51 template<> struct Rmsnorm2dSensitiveEnumName<Rmsnorm2dSensitiveEnum::NO_SPECIFIC_MODEL> { static constexpr const char * name = "nsm"; };
52 template<> struct Rmsnorm2dSensitiveEnumName<Rmsnorm2dSensitiveEnum::T5_MODEL_LIKE> { static constexpr const char * name = "t5ml"; };
Definition: cluster_descriptor.hpp:13
@ NO_SPECIFIC_MODEL
@ T5_MODEL_LIKE
@ SMOOTH_DYNAMIC_QUANT
@ DYNAMIC_QUANT
@ SMOOTH_DYNAMIC_QUANT
@ DYNAMIC_QUANT
@ PRE_ADD_STORE
@ PRE_ADD_STORE
Definition: rmsnorm2d_fwd_traits.hpp:20
Definition: rmsnorm2d_fwd_traits.hpp:34
Definition: rmsnorm2d_fwd_traits.hpp:63
static constexpr bool kSaveUnquant
Definition: rmsnorm2d_fwd_traits.hpp:66
static constexpr Rmsnorm2dFusedAddEnum kFusedAdd
Definition: rmsnorm2d_fwd_traits.hpp:68
static constexpr Rmsnorm2dSensitiveEnum kUseModelSensitiveRMSNorm
Definition: rmsnorm2d_fwd_traits.hpp:70
static constexpr Rmsnorm2dFusedQuantEnum kFusedQuant
Definition: rmsnorm2d_fwd_traits.hpp:69
static constexpr bool kTwoPass
Definition: rmsnorm2d_fwd_traits.hpp:67
static constexpr bool kPadN
Definition: rmsnorm2d_fwd_traits.hpp:64
static constexpr bool kSaveInvRms
Definition: rmsnorm2d_fwd_traits.hpp:65
Definition: rmsnorm2d_fwd_traits.hpp:50