Skip to main content

Mojo module

mha_utils

Aliases

callback_fn_type

alias callback_fn_type = fn[mask_t: MHAMask, score_mod_t: ScoreModTrait](mask: mask_t, score_mod: score_mod_t) raises capturing -> None

is_sm100

alias is_sm100 = String(_accelerator_arch()).__contains__(":100")

is_sm90

alias is_sm90 = String(_accelerator_arch()).__contains__(":90")

is_sm90or100

alias is_sm90or100 = String(_accelerator_arch()).__contains__(":90") if String(_accelerator_arch()).__contains__(":90") else String(_accelerator_arch()).__contains__(":100")

Structs

Traits

Functions

Was this page helpful?