Skip to main content

Mojo struct

KernelConfig

struct KernelConfig

Block/warp/MMA shape configuration for ping-pong kernels.

Fields​

  • ​block_shape (IndexList[3]):
  • ​warp_shape (IndexList[3]):
  • ​mma_shape (IndexList[3]):

Implemented traits​

AnyType, Copyable, ImplicitlyCopyable, ImplicitlyDestructible, Movable, Writable

Methods​

__init__​

__init__(out self, *, block_shape: IndexList[3], warp_shape: IndexList[3], mma_shape: IndexList[3])

num_threads​

num_threads(self) -> Int

Returns:

Int

write_to​

write_to(self, mut writer: T)

write_repr_to​

write_repr_to(self, mut writer: T)