Skip to content
Merged
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
15 changes: 14 additions & 1 deletion cuda_bindings/cuda/bindings/_internal/cufile.pxd
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
#
# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
#
# This code was automatically generated with version 12.9.0. Do not modify it directly.
# This code was automatically generated across versions from 12.9.0 to 13.0.0. Do not modify it directly.

from ..cycufile cimport *

Expand Down Expand Up @@ -41,3 +41,16 @@ cdef CUfileError_t _cuFileGetParameterString(CUFileStringConfigParameter_t param
cdef CUfileError_t _cuFileSetParameterSizeT(CUFileSizeTConfigParameter_t param, size_t value) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
cdef CUfileError_t _cuFileSetParameterBool(CUFileBoolConfigParameter_t param, cpp_bool value) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
cdef CUfileError_t _cuFileSetParameterString(CUFileStringConfigParameter_t param, const char* desc_str) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
cdef CUfileError_t _cuFileDriverClose() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
cdef CUfileError_t _cuFileGetParameterMinMaxValue(CUFileSizeTConfigParameter_t param, size_t* min_value, size_t* max_value) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
cdef CUfileError_t _cuFileSetStatsLevel(int level) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
cdef CUfileError_t _cuFileGetStatsLevel(int* level) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
cdef CUfileError_t _cuFileStatsStart() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
cdef CUfileError_t _cuFileStatsStop() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
cdef CUfileError_t _cuFileStatsReset() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
cdef CUfileError_t _cuFileGetStatsL1(CUfileStatsLevel1_t* stats) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
cdef CUfileError_t _cuFileGetStatsL2(CUfileStatsLevel2_t* stats) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
cdef CUfileError_t _cuFileGetStatsL3(CUfileStatsLevel3_t* stats) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
cdef CUfileError_t _cuFileGetBARSizeInKB(int gpuIndex, size_t* barSize) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
cdef CUfileError_t _cuFileSetParameterPosixPoolSlabArray(const size_t* size_values, const size_t* count_values, int len) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
cdef CUfileError_t _cuFileGetParameterPosixPoolSlabArray(size_t* size_values, size_t* count_values, int len) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
275 changes: 274 additions & 1 deletion cuda_bindings/cuda/bindings/_internal/cufile_linux.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
#
# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
#
# This code was automatically generated with version 12.9.0. Do not modify it directly.
# This code was automatically generated across versions from 12.9.0 to 13.0.0. Do not modify it directly.

from libc.stdint cimport intptr_t, uintptr_t

Expand Down Expand Up @@ -68,6 +68,19 @@ cdef void* __cuFileGetParameterString = NULL
cdef void* __cuFileSetParameterSizeT = NULL
cdef void* __cuFileSetParameterBool = NULL
cdef void* __cuFileSetParameterString = NULL
cdef void* __cuFileDriverClose = NULL
cdef void* __cuFileGetParameterMinMaxValue = NULL
cdef void* __cuFileSetStatsLevel = NULL
cdef void* __cuFileGetStatsLevel = NULL
cdef void* __cuFileStatsStart = NULL
cdef void* __cuFileStatsStop = NULL
cdef void* __cuFileStatsReset = NULL
cdef void* __cuFileGetStatsL1 = NULL
cdef void* __cuFileGetStatsL2 = NULL
cdef void* __cuFileGetStatsL3 = NULL
cdef void* __cuFileGetBARSizeInKB = NULL
cdef void* __cuFileSetParameterPosixPoolSlabArray = NULL
cdef void* __cuFileGetParameterPosixPoolSlabArray = NULL


cdef void* load_library(const int driver_ver) except* with gil:
Expand Down Expand Up @@ -312,6 +325,97 @@ cdef int _check_or_init_cufile() except -1 nogil:
handle = load_library(driver_ver)
__cuFileSetParameterString = dlsym(handle, 'cuFileSetParameterString')

global __cuFileDriverClose
__cuFileDriverClose = dlsym(RTLD_DEFAULT, 'cuFileDriverClose')
if __cuFileDriverClose == NULL:
if handle == NULL:
handle = load_library(driver_ver)
__cuFileDriverClose = dlsym(handle, 'cuFileDriverClose')

global __cuFileGetParameterMinMaxValue
__cuFileGetParameterMinMaxValue = dlsym(RTLD_DEFAULT, 'cuFileGetParameterMinMaxValue')
if __cuFileGetParameterMinMaxValue == NULL:
if handle == NULL:
handle = load_library(driver_ver)
__cuFileGetParameterMinMaxValue = dlsym(handle, 'cuFileGetParameterMinMaxValue')

global __cuFileSetStatsLevel
__cuFileSetStatsLevel = dlsym(RTLD_DEFAULT, 'cuFileSetStatsLevel')
if __cuFileSetStatsLevel == NULL:
if handle == NULL:
handle = load_library(driver_ver)
__cuFileSetStatsLevel = dlsym(handle, 'cuFileSetStatsLevel')

global __cuFileGetStatsLevel
__cuFileGetStatsLevel = dlsym(RTLD_DEFAULT, 'cuFileGetStatsLevel')
if __cuFileGetStatsLevel == NULL:
if handle == NULL:
handle = load_library(driver_ver)
__cuFileGetStatsLevel = dlsym(handle, 'cuFileGetStatsLevel')

global __cuFileStatsStart
__cuFileStatsStart = dlsym(RTLD_DEFAULT, 'cuFileStatsStart')
if __cuFileStatsStart == NULL:
if handle == NULL:
handle = load_library(driver_ver)
__cuFileStatsStart = dlsym(handle, 'cuFileStatsStart')

global __cuFileStatsStop
__cuFileStatsStop = dlsym(RTLD_DEFAULT, 'cuFileStatsStop')
if __cuFileStatsStop == NULL:
if handle == NULL:
handle = load_library(driver_ver)
__cuFileStatsStop = dlsym(handle, 'cuFileStatsStop')

global __cuFileStatsReset
__cuFileStatsReset = dlsym(RTLD_DEFAULT, 'cuFileStatsReset')
if __cuFileStatsReset == NULL:
if handle == NULL:
handle = load_library(driver_ver)
__cuFileStatsReset = dlsym(handle, 'cuFileStatsReset')

global __cuFileGetStatsL1
__cuFileGetStatsL1 = dlsym(RTLD_DEFAULT, 'cuFileGetStatsL1')
if __cuFileGetStatsL1 == NULL:
if handle == NULL:
handle = load_library(driver_ver)
__cuFileGetStatsL1 = dlsym(handle, 'cuFileGetStatsL1')

global __cuFileGetStatsL2
__cuFileGetStatsL2 = dlsym(RTLD_DEFAULT, 'cuFileGetStatsL2')
if __cuFileGetStatsL2 == NULL:
if handle == NULL:
handle = load_library(driver_ver)
__cuFileGetStatsL2 = dlsym(handle, 'cuFileGetStatsL2')

global __cuFileGetStatsL3
__cuFileGetStatsL3 = dlsym(RTLD_DEFAULT, 'cuFileGetStatsL3')
if __cuFileGetStatsL3 == NULL:
if handle == NULL:
handle = load_library(driver_ver)
__cuFileGetStatsL3 = dlsym(handle, 'cuFileGetStatsL3')

global __cuFileGetBARSizeInKB
__cuFileGetBARSizeInKB = dlsym(RTLD_DEFAULT, 'cuFileGetBARSizeInKB')
if __cuFileGetBARSizeInKB == NULL:
if handle == NULL:
handle = load_library(driver_ver)
__cuFileGetBARSizeInKB = dlsym(handle, 'cuFileGetBARSizeInKB')

global __cuFileSetParameterPosixPoolSlabArray
__cuFileSetParameterPosixPoolSlabArray = dlsym(RTLD_DEFAULT, 'cuFileSetParameterPosixPoolSlabArray')
if __cuFileSetParameterPosixPoolSlabArray == NULL:
if handle == NULL:
handle = load_library(driver_ver)
__cuFileSetParameterPosixPoolSlabArray = dlsym(handle, 'cuFileSetParameterPosixPoolSlabArray')

global __cuFileGetParameterPosixPoolSlabArray
__cuFileGetParameterPosixPoolSlabArray = dlsym(RTLD_DEFAULT, 'cuFileGetParameterPosixPoolSlabArray')
if __cuFileGetParameterPosixPoolSlabArray == NULL:
if handle == NULL:
handle = load_library(driver_ver)
__cuFileGetParameterPosixPoolSlabArray = dlsym(handle, 'cuFileGetParameterPosixPoolSlabArray')

__py_cufile_init = True
return 0

Expand Down Expand Up @@ -417,6 +521,45 @@ cpdef dict _inspect_function_pointers():
global __cuFileSetParameterString
data["__cuFileSetParameterString"] = <intptr_t>__cuFileSetParameterString

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

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

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

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

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

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

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

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

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

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

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

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

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

func_ptrs = data
return data

Expand Down Expand Up @@ -732,3 +875,133 @@ cdef CUfileError_t _cuFileSetParameterString(CUFileStringConfigParameter_t param
raise FunctionNotFoundError("function cuFileSetParameterString is not found")
return (<CUfileError_t (*)(CUFileStringConfigParameter_t, const char*) noexcept nogil>__cuFileSetParameterString)(
param, desc_str)


cdef CUfileError_t _cuFileDriverClose() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
global __cuFileDriverClose
_check_or_init_cufile()
if __cuFileDriverClose == NULL:
with gil:
raise FunctionNotFoundError("function cuFileDriverClose is not found")
return (<CUfileError_t (*)() noexcept nogil>__cuFileDriverClose)(
)


cdef CUfileError_t _cuFileGetParameterMinMaxValue(CUFileSizeTConfigParameter_t param, size_t* min_value, size_t* max_value) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
global __cuFileGetParameterMinMaxValue
_check_or_init_cufile()
if __cuFileGetParameterMinMaxValue == NULL:
with gil:
raise FunctionNotFoundError("function cuFileGetParameterMinMaxValue is not found")
return (<CUfileError_t (*)(CUFileSizeTConfigParameter_t, size_t*, size_t*) noexcept nogil>__cuFileGetParameterMinMaxValue)(
param, min_value, max_value)


cdef CUfileError_t _cuFileSetStatsLevel(int level) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
global __cuFileSetStatsLevel
_check_or_init_cufile()
if __cuFileSetStatsLevel == NULL:
with gil:
raise FunctionNotFoundError("function cuFileSetStatsLevel is not found")
return (<CUfileError_t (*)(int) noexcept nogil>__cuFileSetStatsLevel)(
level)


cdef CUfileError_t _cuFileGetStatsLevel(int* level) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
global __cuFileGetStatsLevel
_check_or_init_cufile()
if __cuFileGetStatsLevel == NULL:
with gil:
raise FunctionNotFoundError("function cuFileGetStatsLevel is not found")
return (<CUfileError_t (*)(int*) noexcept nogil>__cuFileGetStatsLevel)(
level)


cdef CUfileError_t _cuFileStatsStart() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
global __cuFileStatsStart
_check_or_init_cufile()
if __cuFileStatsStart == NULL:
with gil:
raise FunctionNotFoundError("function cuFileStatsStart is not found")
return (<CUfileError_t (*)() noexcept nogil>__cuFileStatsStart)(
)


cdef CUfileError_t _cuFileStatsStop() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
global __cuFileStatsStop
_check_or_init_cufile()
if __cuFileStatsStop == NULL:
with gil:
raise FunctionNotFoundError("function cuFileStatsStop is not found")
return (<CUfileError_t (*)() noexcept nogil>__cuFileStatsStop)(
)


cdef CUfileError_t _cuFileStatsReset() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
global __cuFileStatsReset
_check_or_init_cufile()
if __cuFileStatsReset == NULL:
with gil:
raise FunctionNotFoundError("function cuFileStatsReset is not found")
return (<CUfileError_t (*)() noexcept nogil>__cuFileStatsReset)(
)


cdef CUfileError_t _cuFileGetStatsL1(CUfileStatsLevel1_t* stats) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
global __cuFileGetStatsL1
_check_or_init_cufile()
if __cuFileGetStatsL1 == NULL:
with gil:
raise FunctionNotFoundError("function cuFileGetStatsL1 is not found")
return (<CUfileError_t (*)(CUfileStatsLevel1_t*) noexcept nogil>__cuFileGetStatsL1)(
stats)


cdef CUfileError_t _cuFileGetStatsL2(CUfileStatsLevel2_t* stats) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
global __cuFileGetStatsL2
_check_or_init_cufile()
if __cuFileGetStatsL2 == NULL:
with gil:
raise FunctionNotFoundError("function cuFileGetStatsL2 is not found")
return (<CUfileError_t (*)(CUfileStatsLevel2_t*) noexcept nogil>__cuFileGetStatsL2)(
stats)


cdef CUfileError_t _cuFileGetStatsL3(CUfileStatsLevel3_t* stats) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
global __cuFileGetStatsL3
_check_or_init_cufile()
if __cuFileGetStatsL3 == NULL:
with gil:
raise FunctionNotFoundError("function cuFileGetStatsL3 is not found")
return (<CUfileError_t (*)(CUfileStatsLevel3_t*) noexcept nogil>__cuFileGetStatsL3)(
stats)


cdef CUfileError_t _cuFileGetBARSizeInKB(int gpuIndex, size_t* barSize) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
global __cuFileGetBARSizeInKB
_check_or_init_cufile()
if __cuFileGetBARSizeInKB == NULL:
with gil:
raise FunctionNotFoundError("function cuFileGetBARSizeInKB is not found")
return (<CUfileError_t (*)(int, size_t*) noexcept nogil>__cuFileGetBARSizeInKB)(
gpuIndex, barSize)


cdef CUfileError_t _cuFileSetParameterPosixPoolSlabArray(const size_t* size_values, const size_t* count_values, int len) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
global __cuFileSetParameterPosixPoolSlabArray
_check_or_init_cufile()
if __cuFileSetParameterPosixPoolSlabArray == NULL:
with gil:
raise FunctionNotFoundError("function cuFileSetParameterPosixPoolSlabArray is not found")
return (<CUfileError_t (*)(const size_t*, const size_t*, int) noexcept nogil>__cuFileSetParameterPosixPoolSlabArray)(
size_values, count_values, len)


cdef CUfileError_t _cuFileGetParameterPosixPoolSlabArray(size_t* size_values, size_t* count_values, int len) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
global __cuFileGetParameterPosixPoolSlabArray
_check_or_init_cufile()
if __cuFileGetParameterPosixPoolSlabArray == NULL:
with gil:
raise FunctionNotFoundError("function cuFileGetParameterPosixPoolSlabArray is not found")
return (<CUfileError_t (*)(size_t*, size_t*, int) noexcept nogil>__cuFileGetParameterPosixPoolSlabArray)(
size_values, count_values, len)
8 changes: 7 additions & 1 deletion cuda_bindings/cuda/bindings/cufile.pxd
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
#
# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
#
# This code was automatically generated with version 12.9.0. Do not modify it directly.
# This code was automatically generated across versions from 12.9.0 to 13.0.0. Do not modify it directly.

from libc.stdint cimport intptr_t

Expand All @@ -18,7 +18,12 @@ ctypedef CUfileBatchHandle_t BatchHandle
ctypedef CUfileError_t Error
ctypedef cufileRDMAInfo_t RDMAInfo
ctypedef CUfileFSOps_t FSOps
ctypedef CUfileOpCounter_t OpCounter
ctypedef CUfilePerGpuStats_t PerGpuStats
ctypedef CUfileDrvProps_t DrvProps
ctypedef CUfileStatsLevel1_t StatsLevel1
ctypedef CUfileStatsLevel2_t StatsLevel2
ctypedef CUfileStatsLevel3_t StatsLevel3


###############################################################################
Expand All @@ -36,6 +41,7 @@ ctypedef CUfileBatchMode_t _BatchMode
ctypedef CUFileSizeTConfigParameter_t _SizeTConfigParameter
ctypedef CUFileBoolConfigParameter_t _BoolConfigParameter
ctypedef CUFileStringConfigParameter_t _StringConfigParameter
ctypedef CUFileArrayConfigParameter_t _ArrayConfigParameter


###############################################################################
Expand Down
Loading
Loading