Skip to content

Add Python bindings for nvJitLink #157

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

Merged
merged 37 commits into from
Oct 25, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
37 commits
Select commit Hold shift + click to select a range
c6e0a02
naive approach to adding bindings
ksimpson-work Oct 10, 2024
47db0c7
add test file
ksimpson-work Oct 15, 2024
84efbb0
rebase
ksimpson-work Oct 15, 2024
e893cd2
add test file
ksimpson-work Oct 15, 2024
5d60eb1
more changes
ksimpson-work Oct 16, 2024
8c4029f
working
ksimpson-work Oct 16, 2024
8852a92
rebase
ksimpson-work Oct 15, 2024
fb1198a
add test file
ksimpson-work Oct 15, 2024
2e4955f
rebase
ksimpson-work Oct 16, 2024
db42fc7
merging
ksimpson-work Oct 16, 2024
6466003
Merge branch 'main' of github.com:NVIDIA/cuda-python into ksimpson/ad…
ksimpson-work Oct 16, 2024
238736c
rebase merge
ksimpson-work Oct 16, 2024
d4bd29c
cleaning up merge
ksimpson-work Oct 17, 2024
25ba9de
Merge branch 'main' into ksimpson/add_nvJitLink_bindings
leofang Oct 19, 2024
f7f4fe5
regenerate
leofang Oct 19, 2024
990f4cb
clean up a bit
leofang Oct 19, 2024
fa87ea8
strip input enumerator prefix
leofang Oct 19, 2024
5f13031
hand-write nvJitLinkDestroy lowpp binding; turn on doxygen
leofang Oct 19, 2024
f167588
switch from NSTR to improved NSEQ; purge NSTR
leofang Oct 19, 2024
ea3d226
fix inner vector<char> size & clean-up
leofang Oct 19, 2024
c1d21da
use autogen'd error code
leofang Oct 19, 2024
0b836e0
fix input arg conversions & output args; add nvJitLinkVersion
leofang Oct 19, 2024
fab6382
start fixing tests
leofang Oct 20, 2024
7fde00e
clean up a bit
leofang Oct 20, 2024
2298435
add destroy docstring
leofang Oct 21, 2024
cb6c5b4
update tests; regen bindings
ksimpson-work Oct 22, 2024
e941305
update test file TODO add some coverage for GetLinkedPtx, and expecte…
ksimpson-work Oct 22, 2024
df605e9
update test file
ksimpson-work Oct 22, 2024
b1536f3
regenerate bindings
ksimpson-work Oct 22, 2024
992ddcf
regenerate bindings
ksimpson-work Oct 22, 2024
b5c5c1c
regenerate with docstrings
ksimpson-work Oct 22, 2024
8ee6aa2
regenerate bindings with windows imports
ksimpson-work Oct 22, 2024
f276cd6
use tmp_path fixture
leofang Oct 23, 2024
fcdae67
fix license header in the test file
leofang Oct 23, 2024
f861f80
fix nvjitlink dll name
leofang Oct 23, 2024
375c33b
fix typo in setup.py
ksimpson-work Oct 25, 2024
f7704a9
Update cuda_bindings/setup.py
leofang Oct 25, 2024
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Empty file.
27 changes: 27 additions & 0 deletions cuda_bindings/cuda/bindings/_internal/nvjitlink.pxd
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
# Copyright (c) 2024, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
#
# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
#
# This code was automatically generated across versions from 12.0.1 to 12.6.2. Do not modify it directly.

from ..cynvjitlink cimport *


###############################################################################
# Wrapper functions
###############################################################################

cdef nvJitLinkResult _nvJitLinkCreate(nvJitLinkHandle* handle, uint32_t numOptions, const char** options) except* nogil
cdef nvJitLinkResult _nvJitLinkDestroy(nvJitLinkHandle* handle) except* nogil
cdef nvJitLinkResult _nvJitLinkAddData(nvJitLinkHandle handle, nvJitLinkInputType inputType, const void* data, size_t size, const char* name) except* nogil
cdef nvJitLinkResult _nvJitLinkAddFile(nvJitLinkHandle handle, nvJitLinkInputType inputType, const char* fileName) except* nogil
cdef nvJitLinkResult _nvJitLinkComplete(nvJitLinkHandle handle) except* nogil
cdef nvJitLinkResult _nvJitLinkGetLinkedCubinSize(nvJitLinkHandle handle, size_t* size) except* nogil
cdef nvJitLinkResult _nvJitLinkGetLinkedCubin(nvJitLinkHandle handle, void* cubin) except* nogil
cdef nvJitLinkResult _nvJitLinkGetLinkedPtxSize(nvJitLinkHandle handle, size_t* size) except* nogil
cdef nvJitLinkResult _nvJitLinkGetLinkedPtx(nvJitLinkHandle handle, char* ptx) except* nogil
cdef nvJitLinkResult _nvJitLinkGetErrorLogSize(nvJitLinkHandle handle, size_t* size) except* nogil
cdef nvJitLinkResult _nvJitLinkGetErrorLog(nvJitLinkHandle handle, char* log) except* nogil
cdef nvJitLinkResult _nvJitLinkGetInfoLogSize(nvJitLinkHandle handle, size_t* size) except* nogil
cdef nvJitLinkResult _nvJitLinkGetInfoLog(nvJitLinkHandle handle, char* log) except* nogil
cdef nvJitLinkResult _nvJitLinkVersion(unsigned int* major, unsigned int* minor) except* nogil
Loading