Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Implement a Dispatcher object for numba_dpex.kernel decorated functions #1120

Closed
11 of 12 tasks
diptorupd opened this issue Aug 21, 2023 · 1 comment
Closed
11 of 12 tasks
Assignees
Milestone

Comments

@diptorupd
Copy link
Collaborator

diptorupd commented Aug 21, 2023

numba-dpex.kernel decorated functions use the numba_dpex.core.kernel_interface.dispacther.JitKernel class to dispatch kernel functions. The JitKernel class does not derive from Numba's Dispatcher leading to various limitations, such as:

  1. Overloads cannot be defined in the DpexKernelTarget
  2. Kernel functions cannot be called from dpjit
  3. All the kernel submit and unbox/box of arguments logic is in Python.

The issue is to track changing the JitKernel class into a proper Dispatcher object.

List of changes that should be completed:

@diptorupd
Copy link
Collaborator Author

Marking the issue as completed barring the two pending tasks #795 and #1197. The remaining work can be tracked separately from the main task of creating a new Dispatcher for numba_dpex kernel.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant