Mojo struct
SplitKPartition
@register_passable(trivial)
struct SplitKPartition[dtype: DType]
Fields
- ptr (
UnsafePointer[Scalar[dtype]]
): - num_partitions_value (
UInt32
):
Implemented traits
AnyType
,
Copyable
,
ImplicitlyCopyable
,
MHAPartitionScheme
,
Movable
,
UnknownDestructibility
Aliases
__copyinit__is_trivial
alias __copyinit__is_trivial = True
__del__is_trivial
alias __del__is_trivial = True
__moveinit__is_trivial
alias __moveinit__is_trivial = True
accum_dtype
alias accum_dtype = dtype
do_partition
alias do_partition = True
Methods
__init__
__init__(ptr: UnsafePointer[Scalar[dtype]], num_partitions_value: UInt32) -> Self
num_partitions
get_exp_sum_qk_max_pointer
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!