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 = UInt32.__copyinit__is_trivial if UnsafePointer[Scalar[dtype]].__copyinit__is_trivial else UnsafePointer[Scalar[dtype]].__copyinit__is_trivial
__del__is_trivial
alias __del__is_trivial = UInt32.__del__is_trivial if UnsafePointer[Scalar[dtype]].__del__is_trivial else UnsafePointer[Scalar[dtype]].__del__is_trivial
__moveinit__is_trivial
alias __moveinit__is_trivial = UInt32.__moveinit__is_trivial if UnsafePointer[Scalar[dtype]].__moveinit__is_trivial else UnsafePointer[Scalar[dtype]].__moveinit__is_trivial
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!