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__(":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?