Mojo trait
MHAPartitionScheme
Implemented traits
AnyType,
UnknownDestructibility
Aliases
__del__is_trivial
alias __del__is_trivial
A flag (often compiler generated) to indicate whether the implementation of __del__ is trivial.
The implementation of __del__ is considered to be trivial if:
- The struct has a compiler-generated trivial destructor and all its fields
have a trivial
__del__method.
In practice, it means that the __del__ can be considered as no-op.
accum_dtype
alias accum_dtype
do_partition
alias do_partition
Methods
num_partitions
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]]
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!