iron.kernel Namespace Reference

Classes
class	BaseKernel
class	ExternalFunction
class	Kernel

Functions
	_is_contiguous_row_major (mr)
	_maybe_collapse_to_match (arg, expected_ty)

Variables
	logger = logging.getLogger(__name__)

Detailed Description

Kernel and ExternalFunction: wrappers for pre-compiled and C++ AIE compute kernels.

Function Documentation

_is_contiguous_row_major()

iron.kernel._is_contiguous_row_major (   mr )

protected

True iff ``mr`` is fully-static row-major contiguous at offset 0;
required before ``memref.collapse_shape`` (UB on non-contiguous dims).

_maybe_collapse_to_match()

iron.kernel._maybe_collapse_to_match (   arg,   expected_ty  )

protected

Bridge an N-D contiguous memref arg to a 1-D kernel signature via
``memref.collapse_shape``. Iron L1 buffers are multi-dim (e.g.
``memref<64x64xi16>``) but ``aie.iron.kernels.X`` helpers declare
flat 1-D args; without this adapter MLIR rejects the call even though
bytes line up. Aliases storage — no copy emitted. Returns ``arg``
unchanged for any case that isn't safely collapsible, so real bugs
still surface in MLIR verification.

Variable Documentation

logger

iron.kernel.logger = logging.getLogger(__name__)