Mojo struct
NoPartition
struct NoPartition[dtype: DType]
Implemented traits
AnyType,
Copyable,
Defaultable,
ImplicitlyCopyable,
ImplicitlyDestructible,
MHAPartitionScheme,
Movable,
RegisterPassable,
TrivialRegisterPassable
comptime members
accum_dtype
comptime accum_dtype = dtype
do_partition
comptime do_partition = False
Methods
__init__
__init__() -> Self
num_partitions
get_exp_sum_qk_max_pointer
get_exp_sum_qk_max_pointer(self) -> UnsafePointer[Scalar[NoPartition[dtype].accum_dtype], MutAnyOrigin]
Returns:
UnsafePointer[Scalar[NoPartition[dtype].accum_dtype], MutAnyOrigin]
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!