Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Add missing nvvmGetErrorString() bindings. #690

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.

Sign up for GitHub

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 2 commits into from
Jun 7, 2025
Merged

Add missing nvvmGetErrorString() bindings. #690

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
bedaa53
Update generated code to include `nvvmGetErrorString`
rwgk Jun 6, 2025
94884e5
Add test_get_error_string()
rwgk Jun 6, 2025
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
1 change: 1 addition & 0 deletions cuda_bindings/cuda/bindings/_internal/nvvm.pxd
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ from ..cynvvm cimport *
# Wrapper functions
###############################################################################

cdef const char* _nvvmGetErrorString(nvvmResult result) except?NULL nogil
cdef nvvmResult _nvvmVersion(int* major, int* minor) except?_NVVMRESULT_INTERNAL_LOADING_ERROR nogil
cdef nvvmResult _nvvmIRVersion(int* majorIR, int* minorIR, int* majorDbg, int* minorDbg) except?_NVVMRESULT_INTERNAL_LOADING_ERROR nogil
cdef nvvmResult _nvvmCreateProgram(nvvmProgram* prog) except?_NVVMRESULT_INTERNAL_LOADING_ERROR nogil
Expand Down
21 changes: 21 additions & 0 deletions cuda_bindings/cuda/bindings/_internal/nvvm_linux.pyx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,7 @@ cdef extern from "<dlfcn.h>" nogil:
cdef bint __py_nvvm_init = False
cdef void* __cuDriverGetVersion = NULL

cdef void* __nvvmGetErrorString = NULL
cdef void* __nvvmVersion = NULL
cdef void* __nvvmIRVersion = NULL
cdef void* __nvvmCreateProgram = NULL
Expand Down Expand Up @@ -82,6 +83,13 @@ cdef int _check_or_init_nvvm() except -1 nogil:
handle = NULL

# Load function
global __nvvmGetErrorString
__nvvmGetErrorString = dlsym(RTLD_DEFAULT, 'nvvmGetErrorString')
if __nvvmGetErrorString == NULL:
if handle == NULL:
handle = load_library(driver_ver)
__nvvmGetErrorString = dlsym(handle, 'nvvmGetErrorString')

global __nvvmVersion
__nvvmVersion = dlsym(RTLD_DEFAULT, 'nvvmVersion')
if __nvvmVersion == NULL:
Expand Down Expand Up @@ -181,6 +189,9 @@ cpdef dict _inspect_function_pointers():
_check_or_init_nvvm()
cdef dict data = {}

global __nvvmGetErrorString
data["__nvvmGetErrorString"] = <intptr_t>__nvvmGetErrorString

global __nvvmVersion
data["__nvvmVersion"] = <intptr_t>__nvvmVersion

Expand Down Expand Up @@ -232,6 +243,16 @@ cpdef _inspect_function_pointer(str name):
# Wrapper functions
###############################################################################

cdef const char* _nvvmGetErrorString(nvvmResult result) except?NULL nogil:
global __nvvmGetErrorString
_check_or_init_nvvm()
if __nvvmGetErrorString == NULL:
with gil:
raise FunctionNotFoundError("function nvvmGetErrorString is not found")
return (<const char* (*)(nvvmResult) noexcept nogil>__nvvmGetErrorString)(
result)


cdef nvvmResult _nvvmVersion(int* major, int* minor) except?_NVVMRESULT_INTERNAL_LOADING_ERROR nogil:
global __nvvmVersion
_check_or_init_nvvm()
Expand Down
20 changes: 20 additions & 0 deletions cuda_bindings/cuda/bindings/_internal/nvvm_windows.pyx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@ LOAD_LIBRARY_SEARCH_DLL_LOAD_DIR = 0x00000100
cdef bint __py_nvvm_init = False
cdef void* __cuDriverGetVersion = NULL

cdef void* __nvvmGetErrorString = NULL
cdef void* __nvvmVersion = NULL
cdef void* __nvvmIRVersion = NULL
cdef void* __nvvmCreateProgram = NULL
Expand Down Expand Up @@ -62,6 +63,12 @@ cdef int _check_or_init_nvvm() except -1 nogil:
handle = path_finder._load_nvidia_dynamic_library("nvvm").handle

# Load function
global __nvvmGetErrorString
try:
__nvvmGetErrorString = <void*><intptr_t>win32api.GetProcAddress(handle, 'nvvmGetErrorString')
except:
pass

global __nvvmVersion
try:
__nvvmVersion = <void*><intptr_t>win32api.GetProcAddress(handle, 'nvvmVersion')
Expand Down Expand Up @@ -149,6 +156,9 @@ cpdef dict _inspect_function_pointers():
_check_or_init_nvvm()
cdef dict data = {}

global __nvvmGetErrorString
data["__nvvmGetErrorString"] = <intptr_t>__nvvmGetErrorString

global __nvvmVersion
data["__nvvmVersion"] = <intptr_t>__nvvmVersion

Expand Down Expand Up @@ -200,6 +210,16 @@ cpdef _inspect_function_pointer(str name):
# Wrapper functions
###############################################################################

cdef const char* _nvvmGetErrorString(nvvmResult result) except?NULL nogil:
global __nvvmGetErrorString
_check_or_init_nvvm()
if __nvvmGetErrorString == NULL:
with gil:
raise FunctionNotFoundError("function nvvmGetErrorString is not found")
return (<const char* (*)(nvvmResult) noexcept nogil>__nvvmGetErrorString)(
result)


cdef nvvmResult _nvvmVersion(int* major, int* minor) except?_NVVMRESULT_INTERNAL_LOADING_ERROR nogil:
global __nvvmVersion
_check_or_init_nvvm()
Expand Down
1 change: 1 addition & 0 deletions cuda_bindings/cuda/bindings/cynvvm.pxd
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,7 @@ ctypedef void* nvvmProgram 'nvvmProgram'
# Functions
###############################################################################

cdef const char* nvvmGetErrorString(nvvmResult result) except?NULL nogil
cdef nvvmResult nvvmVersion(int* major, int* minor) except?_NVVMRESULT_INTERNAL_LOADING_ERROR nogil
cdef nvvmResult nvvmIRVersion(int* majorIR, int* minorIR, int* majorDbg, int* minorDbg) except?_NVVMRESULT_INTERNAL_LOADING_ERROR nogil
cdef nvvmResult nvvmCreateProgram(nvvmProgram* prog) except?_NVVMRESULT_INTERNAL_LOADING_ERROR nogil
Expand Down
4 changes: 4 additions & 0 deletions cuda_bindings/cuda/bindings/cynvvm.pyx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,10 @@ from ._internal cimport nvvm as _nvvm
# Wrapper functions
###############################################################################

cdef const char* nvvmGetErrorString(nvvmResult result) except?NULL nogil:
return _nvvm._nvvmGetErrorString(result)


cdef nvvmResult nvvmVersion(int* major, int* minor) except?_NVVMRESULT_INTERNAL_LOADING_ERROR nogil:
return _nvvm._nvvmVersion(major, minor)

Expand Down
1 change: 1 addition & 0 deletions cuda_bindings/cuda/bindings/nvvm.pxd
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@ ctypedef nvvmResult _Result
# Functions
###############################################################################

cpdef str get_error_string(int result)
cpdef tuple version()
cpdef tuple ir_version()
cpdef intptr_t create_program() except? 0
Expand Down
13 changes: 13 additions & 0 deletions cuda_bindings/cuda/bindings/nvvm.pyx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -73,6 +73,19 @@ cpdef destroy_program(intptr_t prog):
check_status(status)


cpdef str get_error_string(int result):
"""Get the message string for the given ``nvvmResult`` code.

Args:
result (Result): NVVM API result code.

.. seealso:: `nvvmGetErrorString`
"""
cdef bytes _output_
_output_ = nvvmGetErrorString(<_Result>result)
return _output_.decode()


cpdef tuple version():
"""Get the NVVM version.

Expand Down
14 changes: 14 additions & 0 deletions cuda_bindings/tests/test_nvvm.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -194,6 +194,20 @@ def get_program_log(prog):
return buffer.decode(errors="backslashreplace")


def test_get_error_string():
num_success = 0
num_errors = 0
for enum_obj in nvvm.Result:
es = nvvm.get_error_string(enum_obj)
if enum_obj is nvvm.Result.SUCCESS:
num_success += 1
else:
assert es.startswith("NVVM_ERROR")
num_errors += 1
assert num_success == 1
assert num_errors > 1 # smoke check is sufficient


def test_nvvm_version():
ver = nvvm.version()
assert len(ver) == 2
Expand Down
Loading