iron.kernel.BaseKernel Class Reference

Inheritance diagram for iron.kernel.BaseKernel:

Collaboration diagram for iron.kernel.BaseKernel:

Public Member Functions
	__init__ (self, str name, list[type[np.ndarray]|np.dtype] arg_types=[])
int	tile_size (self, int arg_index=0)
list	arg_types (self)
	__call__ (self, *args, **kwargs)
Public Member Functions inherited from iron.resolvable.Resolvable
None	resolve (self, ir.Location|None loc=None, ir.InsertionPoint|None ip=None)

Protected Attributes
	_name
	_arg_types
	_op

Detailed Description

Base class for AIE core functions that resolve to a func.func declaration.

Subclasses:
    Kernel: wraps a pre-compiled object file.
    ExternalFunction: compiles C/C++ source at JIT time.

Constructor & Destructor Documentation

__init__()

iron.kernel.BaseKernel.__init__	(		self,
		str	name,
		list[type[np.ndarray] | np.dtype]	arg_types = []
	)

Args:
    name: Symbol name of the function.
    arg_types: Type signature of the function arguments.  Defaults to [].

Reimplemented in iron.kernel.Kernel, and iron.kernel.ExternalFunction.

Member Function Documentation

__call__()

iron.kernel.BaseKernel.__call__	(		self,
		*	args,
		**	kwargs
	)

Emit a func.call to this kernel, validating argument count.

Reimplemented in iron.kernel.ExternalFunction.

arg_types()

list iron.kernel.BaseKernel.arg_types

(

self

)

Return a copy of the argument type list.

tile_size()

int iron.kernel.BaseKernel.tile_size	(		self,
		int	arg_index = 0
	)

Return the first dimension of the array argument at ``arg_index``.

Args:
    arg_index: Index into ``arg_types``.  Defaults to 0.

Member Data Documentation

_arg_types

iron.kernel.BaseKernel._arg_types

protected

_name

iron.kernel.BaseKernel._name

protected

_op

iron.kernel.BaseKernel._op

protected

---

The documentation for this class was generated from the following file:

• iron/kernel.py