Mojo trait
MHAPartitionScheme
Implemented traits
AnyType,
UnknownDestructibility
Aliases
accum_dtype
alias accum_dtype
do_partition
alias do_partition
Methods
num_partitions
num_partitions(self: _Self) -> SIMD[uint32, 1]
get_exp_sum_qk_max_pointer
get_exp_sum_qk_max_pointer(self: _Self) -> UnsafePointer[SIMD[get_vtable_entry(:trait<_nn::_mha_sm90::_MHAPartitionScheme> _Self, "accum_dtype"), 1]]
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!