Skip to main content

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_witness(_Self, nn::mha_utils::MHAPartitionScheme, accum_dtype), 1]]

Was this page helpful?