tt-mlir documentation

ArgAttr

Kernel argument.

Syntax:

#ttkernel.arg<
  ArgType,   # arg_type
  size_t,   # operand_index
  bool   # is_uniform
>

Parameters:

ArgSpecAttr

Kernel argument specification.

Syntax:

#ttkernel.arg_spec<
  ::llvm::ArrayRef<ArgAttr>,   # rt_args
  ::llvm::ArrayRef<ArgAttr>   # ct_args
>

A list of argument attibutes, of which form the argument specification for this kernel.

Parameters:

ReduceDimAttr

TTKernel Reduce Dimensions

Syntax:

#ttkernel.reduce_dim<
  ::mlir::tt::ttkernel::ReduceDim   # value
>

Enum cases:

• reduce_dim_row (Row)
• reduce_dim_col (Col)
• reduce_dim_scalar (Scalar)
• reduce_dim_none (None)

Parameters:

ReduceTypeAttr

TTKernel Reduce Types

Syntax:

#ttkernel.reduce_type<
  ::mlir::tt::ttkernel::ReduceType   # value
>

Enum cases:

• reduce_sum (Sum)
• reduce_max (Max)

Parameters:

ThreadTypeAttr

TTKernel ThreadTypes

Syntax:

#ttkernel.thread<
  ::mlir::tt::ttkernel::ThreadType   # value
>

Enum cases:

• noc (Noc)
• compute (Compute)
• ethernet (Ethernet)

Parameters: