Skip to content

TTMetalOp

Nicholas Smith edited this page Jun 24, 2024 · 1 revision

ttmetal.alloc (tt::ttmetal::AllocOp)

Alloc op.

Tensor Alloc operation

Attributes:

Attribute MLIR Type Description
address ::mlir::IntegerAttr 64-bit signless integer attribute
size ::mlir::IntegerAttr 64-bit signless integer attribute
memory_space ::mlir::tt::MemorySpaceAttr
TT MemorySpace{{% markdown %}}Enum cases: * system (`System`) * mmio (`SystemMMIO`) * dram (`DeviceDRAM`) * l1 (`DeviceL1`){{% /markdown %}}

Results:

Result Description
result ranked tensor of any type values

ttmetal.dealloc (tt::ttmetal::DeallocOp)

Dealloc op.

Tensor Dealloc operation

Operands:

Operand Description
input ranked tensor of any type values

ttmetal.dispatch (tt::ttmetal::DispatchOp)

Dispatch op.

Dispatch operation

Traits: AttrSizedOperandSegments

Interfaces: DestinationStyleOpInterface

Attributes:

Attribute MLIR Type Description
core_ranges ::mlir::ArrayAttr
threadTypes ::mlir::ArrayAttr
operand_cb_port_mapping ::mlir::ArrayAttr array attribute

Operands:

Operand Description
inputs variadic of ranked tensor of any type values
outputs variadic of ranked tensor of any type values

Results:

Result Description
results variadic of ranked tensor of any type values

ttmetal.host_read (tt::ttmetal::HostReadOp)

Host read op.

Host read operation

Interfaces: DestinationStyleOpInterface

Operands:

Operand Description
input ranked tensor of any type values
output ranked tensor of any type values

Results:

Result Description
result ranked tensor of any type values

ttmetal.host_write (tt::ttmetal::HostWriteOp)

Host write op.

Host write operation

Interfaces: DestinationStyleOpInterface

Operands:

Operand Description
input ranked tensor of any type values
output ranked tensor of any type values

Results:

Result Description
result ranked tensor of any type values