Skip to main content

Mojo struct

TMALoadOpArgs

struct TMALoadOpArgs[a_type: DType, b_type: DType, a_layout: Layout, b_layout: Layout, a_desc_layout: Layout = a_layout, b_desc_layout: Layout = b_layout]

Fields

  • a_tma_op (TMATensorTile[a_type, a_layout, a_desc_layout]):
  • b_tma_op (TMATensorTile[b_type, b_layout, b_desc_layout]):

Implemented traits

AnyType, Copyable, OpArgs, UnknownDestructibility

Methods

__init__

__init__(out self, a: TMATensorTile[a_type, a_layout, a_desc_layout], b: TMATensorTile[b_type, b_layout, b_desc_layout])

__copyinit__

__copyinit__(out self, other: Self)

Was this page helpful?