iron.jit.NPUKernel Class Reference

Public Member Functions
	__init__ (self, xclbin_path, insts_path, device_index=0, kernel_name="PP_FD_PRE")
	__call__ (self, *args)
	__del__ (self)

Private Attributes
	__device
	__xclbin
	__context
	__kernel
	__n_insts
	__insts_buffer_bo

Detailed Description

NPUKernel class wrapper for NPU kernels.

Constructor & Destructor Documentation

__init__()

iron.jit.NPUKernel.__init__	(		self,
			xclbin_path,
			insts_path,
			device_index = 0,
			kernel_name = "PP_FD_PRE"
	)

Initialize the NPUKernel object.
Parameters:
    xclbin_path (str): Path to the XCLBIN file containing the kernel.
    insts_path (str): Path to the instruction binary file for the kernel.
    device_index (int, optional): Index of the device. Defaults to 0.
    kernel_name (str, optional): Name of the kernel. Defaults to "PP_FD_PRE".

__del__()

iron.jit.NPUKernel.__del__

(

self

)

Destructor to clean up resources and delete the kernel and device objects.

Member Function Documentation

__call__()

iron.jit.NPUKernel.__call__	(		self,
		*	args
	)

Allows the kernel to be called as a function with the provided arguments.

Parameters:
    args (IRON Tensors): Arguments to pass to the kernel.

Member Data Documentation

__context

iron.jit.NPUKernel.__context

private

__device

iron.jit.NPUKernel.__device

private

__insts_buffer_bo

iron.jit.NPUKernel.__insts_buffer_bo

private

__kernel

iron.jit.NPUKernel.__kernel

private

__n_insts

iron.jit.NPUKernel.__n_insts

private

__xclbin

iron.jit.NPUKernel.__xclbin

private

---

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

• iron/jit.py