'npu_schedule' Dialect

NPU scheduled pipeline dialect Public NPU layer that makes scheduling decisions explicit: tile boundaries, engine assignment, and async dependencies.

Operations

npu_schedule.dma_start (iree_compiler::NPUSchedule::DmaStartOp)

Scheduled async DMA launch

Syntax:

operation ::= `npu_schedule.dma_start` $src `,` $dst `size` `=` $size `src_space` `=` $src_space `dst_space` `=` $dst_space
              attr-dict `:` type($src) `to` type($dst) `->` type($token)

Attributes:

Operands:

Results:

npu_schedule.dma_wait (iree_compiler::NPUSchedule::DmaWaitOp)

Scheduled wait on DMA token

Syntax:

operation ::= `npu_schedule.dma_wait` $token attr-dict `:` type($token)

Operands:

npu_schedule.matmul_tile (iree_compiler::NPUSchedule::MatmulTileOp)

Scheduled matmul tile

Syntax:

operation ::= `npu_schedule.matmul_tile` $lhs `,` $rhs attr-dict `:` type($lhs) `,` type($rhs) `->` type($result)

Operands:

Results:

npu_schedule.softmax_fragment (iree_compiler::NPUSchedule::SoftmaxFragmentOp)

Scheduled softmax vector fragment

Syntax:

operation ::= `npu_schedule.softmax_fragment` $input attr-dict `:` type($input) `->` type($result)

Operands:

Results:

npu_schedule.ukernel_launch (iree_compiler::NPUSchedule::UKernelLaunchOp)

Scheduled ukernel launch

Syntax:

operation ::= `npu_schedule.ukernel_launch` $symbol `(` $inputs `)` attr-dict `:` type($inputs) `->` type($result)

Attributes:

Operands:

Results: