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

[FEA] Support JITting and Linking LTOIR from cuda source inputs #45

Open
isVoid opened this issue Aug 23, 2024 · 1 comment
Open

[FEA] Support JITting and Linking LTOIR from cuda source inputs #45

isVoid opened this issue Aug 23, 2024 · 1 comment
Labels
feature request New feature or request
Milestone

Comments

@isVoid
Copy link

isVoid commented Aug 23, 2024

Is your feature request related to a problem? Please describe.
After Pynvjitlink is introduced into Numba-CUDA, I would like to be able to jit cuda source files into LTOIR and link it with nvjitlink. This basically requires the linker lto flag to be respected by the jit and linker machinery.

Describe the solution you'd like
Add lto flag to nvrtc.compile, and add -dlto flag to compile option when set. In the Linker class, use add_ltoir in add_cu when self.lto is set.

@isVoid isVoid added the feature request New feature or request label Aug 23, 2024
@isVoid
Copy link
Author

isVoid commented Aug 23, 2024

As a follow up to this PR - it would be helpful to support -ptx flag for nvjitlink to emit the PTX under LTOIR linkage, given NUMBA_DUMP_ASSEMBLY is set. This makes sure that Numba-CUDA shows the LTO-ed PTX and is more useful in debugging optimization related issues.

@gmarkall gmarkall added this to the v0.0.18 milestone Oct 21, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature request New feature or request
Projects
None yet
Development

No branches or pull requests

2 participants