Skip to main content
Log in

Mojo struct

NoPartition

@register_passable(trivial) struct NoPartition[dtype: DType]

Aliases

  • do_partition = False:
  • accum_dtype = dtype:

Implemented traits

AnyType, Copyable, ExplicitlyCopyable, MHAPartitionScheme, Movable, UnknownDestructibility

Methods

__init__

__init__() -> Self

num_partitions

num_partitions(self) -> SIMD[uint32, 1]

get_exp_sum_qk_max_pointer

get_exp_sum_qk_max_pointer(self) -> UnsafePointer[SIMD[dtype, 1]]

Was this page helpful?