Skip to main content

/

Mojo struct

NoPartition

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

Implemented traits

AnyType, Copyable, Defaultable, MHAPartitionScheme, Movable, UnknownDestructibility

Aliases

`accum_dtype`

alias accum_dtype = dtype

`do_partition`

alias do_partition = False

Methods

`init`

__init__() -> Self

`num_partitions`

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

Returns:

`get_exp_sum_qk_max_pointer`

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

Returns:

UnsafePointer

Implemented traits
Aliases
- accum_dtype
- do_partition
Methods

View source

View source

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!