Npu kernel ops

npu_kernel.dma_await (iree_compiler::NPUKernel::DmaAwaitOp)

Wait on async DMA token

Syntax:

operation ::= `npu_kernel.dma_await` $token attr-dict `:` type($token)

Operands:

npu_kernel.dma_start (iree_compiler::NPUKernel::DmaStartOp)

Async DMA launch returning an async token

Syntax:

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

Attributes:

Operands:

Results:

npu_kernel.matmul (iree_compiler::NPUKernel::MatmulOp)

NPU kernel-level matmul intent

Syntax:

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

Operands:

Results:

npu_kernel.ukernel_generic (iree_compiler::NPUKernel::UKernelGenericOp)

Kernel-level generic ukernel call

Syntax:

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

Attributes:

Operands:

Results: