Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
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
8 changes: 8 additions & 0 deletions cuda_bindings/regenerate_abi.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
#!/usr/bin/env bash

# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
#
# SPDX-License-Identifier: Apache-2.0

SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
python ${SCRIPT_DIR}/tests/test_cython_abi.py regenerate
472 changes: 472 additions & 0 deletions cuda_bindings/tests/abi/cuda/bindings/_bindings/cydriver.abi.json

Large diffs are not rendered by default.

26 changes: 26 additions & 0 deletions cuda_bindings/tests/abi/cuda/bindings/_bindings/cynvrtc.abi.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
{
"_nvrtcAddNameExpression": "nvrtcResult (nvrtcProgram, char const *)",
"_nvrtcCompileProgram": "nvrtcResult (nvrtcProgram, int, char const **)",
"_nvrtcCreateProgram": "nvrtcResult (nvrtcProgram *, char const *, char const *, int, char const **, char const **)",
"_nvrtcDestroyProgram": "nvrtcResult (nvrtcProgram *)",
"_nvrtcGetCUBIN": "nvrtcResult (nvrtcProgram, char *)",
"_nvrtcGetCUBINSize": "nvrtcResult (nvrtcProgram, size_t *)",
"_nvrtcGetErrorString": "char const *(nvrtcResult)",
"_nvrtcGetLTOIR": "nvrtcResult (nvrtcProgram, char *)",
"_nvrtcGetLTOIRSize": "nvrtcResult (nvrtcProgram, size_t *)",
"_nvrtcGetLoweredName": "nvrtcResult (nvrtcProgram, char const *, char const **)",
"_nvrtcGetNumSupportedArchs": "nvrtcResult (int *)",
"_nvrtcGetOptiXIR": "nvrtcResult (nvrtcProgram, char *)",
"_nvrtcGetOptiXIRSize": "nvrtcResult (nvrtcProgram, size_t *)",
"_nvrtcGetPCHCreateStatus": "nvrtcResult (nvrtcProgram)",
"_nvrtcGetPCHHeapSize": "nvrtcResult (size_t *)",
"_nvrtcGetPCHHeapSizeRequired": "nvrtcResult (nvrtcProgram, size_t *)",
"_nvrtcGetPTX": "nvrtcResult (nvrtcProgram, char *)",
"_nvrtcGetPTXSize": "nvrtcResult (nvrtcProgram, size_t *)",
"_nvrtcGetProgramLog": "nvrtcResult (nvrtcProgram, char *)",
"_nvrtcGetProgramLogSize": "nvrtcResult (nvrtcProgram, size_t *)",
"_nvrtcGetSupportedArchs": "nvrtcResult (int *)",
"_nvrtcSetFlowCallback": "nvrtcResult (nvrtcProgram, void *, void *)",
"_nvrtcSetPCHHeapSize": "nvrtcResult (size_t)",
"_nvrtcVersion": "nvrtcResult (int *, int *)"
}
317 changes: 317 additions & 0 deletions cuda_bindings/tests/abi/cuda/bindings/_bindings/cyruntime.abi.json

Large diffs are not rendered by default.

Large diffs are not rendered by default.

45 changes: 45 additions & 0 deletions cuda_bindings/tests/abi/cuda/bindings/_internal/cufile.abi.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
{
"_cuFileBatchIOCancel": "CUfileError_t (CUfileBatchHandle_t)",
"_cuFileBatchIODestroy": "void (CUfileBatchHandle_t)",
"_cuFileBatchIOGetStatus": "CUfileError_t (CUfileBatchHandle_t, unsigned int, unsigned int *, CUfileIOEvents_t *, struct timespec *)",
"_cuFileBatchIOSetUp": "CUfileError_t (CUfileBatchHandle_t *, unsigned int)",
"_cuFileBatchIOSubmit": "CUfileError_t (CUfileBatchHandle_t, unsigned int, CUfileIOParams_t *, unsigned int)",
"_cuFileBufDeregister": "CUfileError_t (void const *)",
"_cuFileBufRegister": "CUfileError_t (void const *, size_t, int)",
"_cuFileDriverClose": "CUfileError_t (void)",
"_cuFileDriverClose_v2": "CUfileError_t (void)",
"_cuFileDriverGetProperties": "CUfileError_t (CUfileDrvProps_t *)",
"_cuFileDriverOpen": "CUfileError_t (void)",
"_cuFileDriverSetMaxCacheSize": "CUfileError_t (size_t)",
"_cuFileDriverSetMaxDirectIOSize": "CUfileError_t (size_t)",
"_cuFileDriverSetMaxPinnedMemSize": "CUfileError_t (size_t)",
"_cuFileDriverSetPollMode": "CUfileError_t (bool, size_t)",
"_cuFileGetBARSizeInKB": "CUfileError_t (int, size_t *)",
"_cuFileGetParameterBool": "CUfileError_t (CUFileBoolConfigParameter_t, bool *)",
"_cuFileGetParameterMinMaxValue": "CUfileError_t (CUFileSizeTConfigParameter_t, size_t *, size_t *)",
"_cuFileGetParameterPosixPoolSlabArray": "CUfileError_t (size_t *, size_t *, int)",
"_cuFileGetParameterSizeT": "CUfileError_t (CUFileSizeTConfigParameter_t, size_t *)",
"_cuFileGetParameterString": "CUfileError_t (CUFileStringConfigParameter_t, char *, int)",
"_cuFileGetStatsL1": "CUfileError_t (CUfileStatsLevel1_t *)",
"_cuFileGetStatsL2": "CUfileError_t (CUfileStatsLevel2_t *)",
"_cuFileGetStatsL3": "CUfileError_t (CUfileStatsLevel3_t *)",
"_cuFileGetStatsLevel": "CUfileError_t (int *)",
"_cuFileGetVersion": "CUfileError_t (int *)",
"_cuFileHandleDeregister": "void (CUfileHandle_t)",
"_cuFileHandleRegister": "CUfileError_t (CUfileHandle_t *, CUfileDescr_t *)",
"_cuFileRead": "Py_ssize_t (CUfileHandle_t, void *, size_t, off_t, off_t)",
"_cuFileReadAsync": "CUfileError_t (CUfileHandle_t, void *, size_t *, off_t *, off_t *, Py_ssize_t *, CUstream)",
"_cuFileSetParameterBool": "CUfileError_t (CUFileBoolConfigParameter_t, bool)",
"_cuFileSetParameterPosixPoolSlabArray": "CUfileError_t (size_t const *, size_t const *, int)",
"_cuFileSetParameterSizeT": "CUfileError_t (CUFileSizeTConfigParameter_t, size_t)",
"_cuFileSetParameterString": "CUfileError_t (CUFileStringConfigParameter_t, char const *)",
"_cuFileSetStatsLevel": "CUfileError_t (int)",
"_cuFileStatsReset": "CUfileError_t (void)",
"_cuFileStatsStart": "CUfileError_t (void)",
"_cuFileStatsStop": "CUfileError_t (void)",
"_cuFileStreamDeregister": "CUfileError_t (CUstream)",
"_cuFileStreamRegister": "CUfileError_t (CUstream, unsigned int)",
"_cuFileUseCount": "long (void)",
"_cuFileWrite": "Py_ssize_t (CUfileHandle_t, void const *, size_t, off_t, off_t)",
"_cuFileWriteAsync": "CUfileError_t (CUfileHandle_t, void *, size_t *, off_t *, off_t *, Py_ssize_t *, CUstream)"
}
16 changes: 16 additions & 0 deletions cuda_bindings/tests/abi/cuda/bindings/_internal/nvjitlink.abi.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
{
"_nvJitLinkAddData": "nvJitLinkResult (nvJitLinkHandle, nvJitLinkInputType, void const *, size_t, char const *)",
"_nvJitLinkAddFile": "nvJitLinkResult (nvJitLinkHandle, nvJitLinkInputType, char const *)",
"_nvJitLinkComplete": "nvJitLinkResult (nvJitLinkHandle)",
"_nvJitLinkCreate": "nvJitLinkResult (nvJitLinkHandle *, uint32_t, char const **)",
"_nvJitLinkDestroy": "nvJitLinkResult (nvJitLinkHandle *)",
"_nvJitLinkGetErrorLog": "nvJitLinkResult (nvJitLinkHandle, char *)",
"_nvJitLinkGetErrorLogSize": "nvJitLinkResult (nvJitLinkHandle, size_t *)",
"_nvJitLinkGetInfoLog": "nvJitLinkResult (nvJitLinkHandle, char *)",
"_nvJitLinkGetInfoLogSize": "nvJitLinkResult (nvJitLinkHandle, size_t *)",
"_nvJitLinkGetLinkedCubin": "nvJitLinkResult (nvJitLinkHandle, void *)",
"_nvJitLinkGetLinkedCubinSize": "nvJitLinkResult (nvJitLinkHandle, size_t *)",
"_nvJitLinkGetLinkedPtx": "nvJitLinkResult (nvJitLinkHandle, char *)",
"_nvJitLinkGetLinkedPtxSize": "nvJitLinkResult (nvJitLinkHandle, size_t *)",
"_nvJitLinkVersion": "nvJitLinkResult (unsigned int *, unsigned int *)"
}
15 changes: 15 additions & 0 deletions cuda_bindings/tests/abi/cuda/bindings/_internal/nvvm.abi.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
{
"_nvvmAddModuleToProgram": "nvvmResult (nvvmProgram, char const *, size_t, char const *)",
"_nvvmCompileProgram": "nvvmResult (nvvmProgram, int, char const **)",
"_nvvmCreateProgram": "nvvmResult (nvvmProgram *)",
"_nvvmDestroyProgram": "nvvmResult (nvvmProgram *)",
"_nvvmGetCompiledResult": "nvvmResult (nvvmProgram, char *)",
"_nvvmGetCompiledResultSize": "nvvmResult (nvvmProgram, size_t *)",
"_nvvmGetErrorString": "char const *(nvvmResult)",
"_nvvmGetProgramLog": "nvvmResult (nvvmProgram, char *)",
"_nvvmGetProgramLogSize": "nvvmResult (nvvmProgram, size_t *)",
"_nvvmIRVersion": "nvvmResult (int *, int *, int *, int *)",
"_nvvmLazyAddModuleToProgram": "nvvmResult (nvvmProgram, char const *, size_t, char const *)",
"_nvvmVerifyProgram": "nvvmResult (nvvmProgram, int, char const **)",
"_nvvmVersion": "nvvmResult (int *, int *)"
}
17 changes: 17 additions & 0 deletions cuda_bindings/tests/abi/cuda/bindings/_internal/utils.abi.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
{
"__pyx_fuse_0get_nested_resource_ptr": "int (__pyx_t_4cuda_8bindings_9_internal_5utils_nested_resource<int> &, PyObject *, int *)",
"__pyx_fuse_0get_resource_ptr": "int (nullable_unique_ptr<std::vector<int> > &, PyObject *, int *)",
"__pyx_fuse_0get_resource_ptrs": "int (nullable_unique_ptr<std::vector<void *> > &, PyObject *, void *)",
"__pyx_fuse_1get_nested_resource_ptr": "int (__pyx_t_4cuda_8bindings_9_internal_5utils_nested_resource<int32_t> &, PyObject *, int32_t *)",
"__pyx_fuse_1get_resource_ptr": "int (nullable_unique_ptr<std::vector<int32_t> > &, PyObject *, int32_t *)",
"__pyx_fuse_2get_nested_resource_ptr": "int (__pyx_t_4cuda_8bindings_9_internal_5utils_nested_resource<int64_t> &, PyObject *, int64_t *)",
"__pyx_fuse_2get_resource_ptr": "int (nullable_unique_ptr<std::vector<int64_t> > &, PyObject *, int64_t *)",
"__pyx_fuse_3get_nested_resource_ptr": "int (__pyx_t_4cuda_8bindings_9_internal_5utils_nested_resource<char> &, PyObject *, char *)",
"__pyx_fuse_3get_resource_ptr": "int (nullable_unique_ptr<std::vector<char> > &, PyObject *, char *)",
"__pyx_fuse_4get_nested_resource_ptr": "int (__pyx_t_4cuda_8bindings_9_internal_5utils_nested_resource<float> &, PyObject *, float *)",
"__pyx_fuse_4get_resource_ptr": "int (nullable_unique_ptr<std::vector<float> > &, PyObject *, float *)",
"__pyx_fuse_5get_nested_resource_ptr": "int (__pyx_t_4cuda_8bindings_9_internal_5utils_nested_resource<double> &, PyObject *, double *)",
"__pyx_fuse_5get_resource_ptr": "int (nullable_unique_ptr<std::vector<double> > &, PyObject *, double *)",
"get_buffer_pointer": "void *(PyObject *, Py_ssize_t, struct __pyx_opt_args_4cuda_8bindings_9_internal_5utils_get_buffer_pointer *__pyx_optional_args)",
"is_nested_sequence": "int (PyObject *)"
}
31 changes: 31 additions & 0 deletions cuda_bindings/tests/abi/cuda/bindings/cufile.abi.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
{
"batch_io_cancel": "PyObject *(intptr_t, int __pyx_skip_dispatch)",
"batch_io_destroy": "void (intptr_t, int __pyx_skip_dispatch)",
"batch_io_get_status": "PyObject *(intptr_t, unsigned int, intptr_t, intptr_t, intptr_t, int __pyx_skip_dispatch)",
"batch_io_set_up": "intptr_t (unsigned int, int __pyx_skip_dispatch)",
"batch_io_submit": "PyObject *(intptr_t, unsigned int, intptr_t, unsigned int, int __pyx_skip_dispatch)",
"buf_deregister": "PyObject *(intptr_t, int __pyx_skip_dispatch)",
"buf_register": "PyObject *(intptr_t, size_t, int, int __pyx_skip_dispatch)",
"driver_get_properties": "PyObject *(intptr_t, int __pyx_skip_dispatch)",
"driver_open": "PyObject *(int __pyx_skip_dispatch)",
"driver_set_max_cache_size": "PyObject *(size_t, int __pyx_skip_dispatch)",
"driver_set_max_direct_io_size": "PyObject *(size_t, int __pyx_skip_dispatch)",
"driver_set_max_pinned_mem_size": "PyObject *(size_t, int __pyx_skip_dispatch)",
"driver_set_poll_mode": "PyObject *(int, size_t, int __pyx_skip_dispatch)",
"get_parameter_bool": "int (int, int __pyx_skip_dispatch)",
"get_parameter_size_t": "size_t (int, int __pyx_skip_dispatch)",
"get_parameter_string": "PyObject *(int, int, int __pyx_skip_dispatch)",
"get_version": "int (int __pyx_skip_dispatch)",
"handle_deregister": "void (intptr_t, int __pyx_skip_dispatch)",
"handle_register": "intptr_t (intptr_t, int __pyx_skip_dispatch)",
"read": "PyObject *(intptr_t, intptr_t, size_t, off_t, off_t, int __pyx_skip_dispatch)",
"read_async": "PyObject *(intptr_t, intptr_t, intptr_t, intptr_t, intptr_t, intptr_t, intptr_t, int __pyx_skip_dispatch)",
"set_parameter_bool": "PyObject *(int, int, int __pyx_skip_dispatch)",
"set_parameter_size_t": "PyObject *(int, size_t, int __pyx_skip_dispatch)",
"set_parameter_string": "PyObject *(int, intptr_t, int __pyx_skip_dispatch)",
"stream_deregister": "PyObject *(intptr_t, int __pyx_skip_dispatch)",
"stream_register": "PyObject *(intptr_t, unsigned int, int __pyx_skip_dispatch)",
"use_count": "PyObject *(int __pyx_skip_dispatch)",
"write": "PyObject *(intptr_t, intptr_t, size_t, off_t, off_t, int __pyx_skip_dispatch)",
"write_async": "PyObject *(intptr_t, intptr_t, intptr_t, intptr_t, intptr_t, intptr_t, intptr_t, int __pyx_skip_dispatch)"
}
45 changes: 45 additions & 0 deletions cuda_bindings/tests/abi/cuda/bindings/cycufile.abi.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
{
"cuFileBatchIOCancel": "CUfileError_t (CUfileBatchHandle_t)",
"cuFileBatchIODestroy": "void (CUfileBatchHandle_t)",
"cuFileBatchIOGetStatus": "CUfileError_t (CUfileBatchHandle_t, unsigned int, unsigned int *, CUfileIOEvents_t *, struct timespec *)",
"cuFileBatchIOSetUp": "CUfileError_t (CUfileBatchHandle_t *, unsigned int)",
"cuFileBatchIOSubmit": "CUfileError_t (CUfileBatchHandle_t, unsigned int, CUfileIOParams_t *, unsigned int)",
"cuFileBufDeregister": "CUfileError_t (void const *)",
"cuFileBufRegister": "CUfileError_t (void const *, size_t, int)",
"cuFileDriverClose": "CUfileError_t (void)",
"cuFileDriverClose_v2": "CUfileError_t (void)",
"cuFileDriverGetProperties": "CUfileError_t (CUfileDrvProps_t *)",
"cuFileDriverOpen": "CUfileError_t (void)",
"cuFileDriverSetMaxCacheSize": "CUfileError_t (size_t)",
"cuFileDriverSetMaxDirectIOSize": "CUfileError_t (size_t)",
"cuFileDriverSetMaxPinnedMemSize": "CUfileError_t (size_t)",
"cuFileDriverSetPollMode": "CUfileError_t (bool, size_t)",
"cuFileGetBARSizeInKB": "CUfileError_t (int, size_t *)",
"cuFileGetParameterBool": "CUfileError_t (CUFileBoolConfigParameter_t, bool *)",
"cuFileGetParameterMinMaxValue": "CUfileError_t (CUFileSizeTConfigParameter_t, size_t *, size_t *)",
"cuFileGetParameterPosixPoolSlabArray": "CUfileError_t (size_t *, size_t *, int)",
"cuFileGetParameterSizeT": "CUfileError_t (CUFileSizeTConfigParameter_t, size_t *)",
"cuFileGetParameterString": "CUfileError_t (CUFileStringConfigParameter_t, char *, int)",
"cuFileGetStatsL1": "CUfileError_t (CUfileStatsLevel1_t *)",
"cuFileGetStatsL2": "CUfileError_t (CUfileStatsLevel2_t *)",
"cuFileGetStatsL3": "CUfileError_t (CUfileStatsLevel3_t *)",
"cuFileGetStatsLevel": "CUfileError_t (int *)",
"cuFileGetVersion": "CUfileError_t (int *)",
"cuFileHandleDeregister": "void (CUfileHandle_t)",
"cuFileHandleRegister": "CUfileError_t (CUfileHandle_t *, CUfileDescr_t *)",
"cuFileRead": "Py_ssize_t (CUfileHandle_t, void *, size_t, off_t, off_t)",
"cuFileReadAsync": "CUfileError_t (CUfileHandle_t, void *, size_t *, off_t *, off_t *, Py_ssize_t *, CUstream)",
"cuFileSetParameterBool": "CUfileError_t (CUFileBoolConfigParameter_t, bool)",
"cuFileSetParameterPosixPoolSlabArray": "CUfileError_t (size_t const *, size_t const *, int)",
"cuFileSetParameterSizeT": "CUfileError_t (CUFileSizeTConfigParameter_t, size_t)",
"cuFileSetParameterString": "CUfileError_t (CUFileStringConfigParameter_t, char const *)",
"cuFileSetStatsLevel": "CUfileError_t (int)",
"cuFileStatsReset": "CUfileError_t (void)",
"cuFileStatsStart": "CUfileError_t (void)",
"cuFileStatsStop": "CUfileError_t (void)",
"cuFileStreamDeregister": "CUfileError_t (CUstream)",
"cuFileStreamRegister": "CUfileError_t (CUstream, unsigned int)",
"cuFileUseCount": "long (void)",
"cuFileWrite": "Py_ssize_t (CUfileHandle_t, void const *, size_t, off_t, off_t)",
"cuFileWriteAsync": "CUfileError_t (CUfileHandle_t, void *, size_t *, off_t *, off_t *, Py_ssize_t *, CUstream)"
}
Loading
Loading