Skip to main content
Log in

Mojo trait

MHAPartitionScheme

Implemented traits

AnyType, UnknownDestructibility

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]]