Skip to main content

Mojo module

mha_utils

Aliases

callback_fn_type

alias callback_fn_type = fn[MHAMask, ScoreModTrait](mask: $0, score_mod: $1) raises capturing -> None

is_sm100

alias is_sm100 = String(_accelerator_arch()).__contains__[::Bool,::Origin[$0]](":100")

is_sm90

alias is_sm90 = String(_accelerator_arch()).__contains__[::Bool,::Origin[$0]](":90")

is_sm90or100

alias is_sm90or100 = String(_accelerator_arch()).__contains__[::Bool,::Origin[$0]](":90") if String(_accelerator_arch()).__contains__[::Bool,::Origin[$0]](":90") else String(_accelerator_arch()).__contains__[::Bool,::Origin[$0]](":100")

Structs

Traits

Functions

Was this page helpful?