Skip to content

Prepare for CTK 13.4 #2423

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

Draft
mdboom wants to merge 1 commit into
base: main
Choose a base branch
from
Draft

Prepare for CTK 13.4 #2423

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
6 changes: 5 additions & 1 deletion cuda_bindings/cuda/bindings/_internal/nvml.pxd
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
#
# This code was automatically generated across versions from 12.9.1 to 13.3.0. Do not modify it directly.

# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=ed4c433854399c2e4f1adc3ffcfc37901e1be29c5aa50728498c87225edf92b1
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=c2cc3cd086b5aeea5fad7ca17600d0102691a3cb354b916f5c086c383d77df19
from ..cynvml cimport *


Expand Down Expand Up @@ -363,3 +363,7 @@ cdef nvmlReturn_t _nvmlDeviceGetVgpuSchedulerLog_v2(nvmlDevice_t device, nvmlVgp
cdef nvmlReturn_t _nvmlGpuInstanceGetVgpuSchedulerLog_v2(nvmlGpuInstance_t gpuInstance, nvmlVgpuSchedulerLogInfo_v2_t* pSchedulerLogInfo) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil
cdef nvmlReturn_t _nvmlDeviceSetVgpuSchedulerState_v2(nvmlDevice_t device, nvmlVgpuSchedulerState_v2_t* pSchedulerState) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil
cdef nvmlReturn_t _nvmlGpuInstanceSetVgpuSchedulerState_v2(nvmlGpuInstance_t gpuInstance, nvmlVgpuSchedulerState_v2_t* pSchedulerState) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil
cdef nvmlReturn_t _nvmlSystemGetCPER_v1(nvmlGetCPER_v1_t* cper) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil
cdef nvmlReturn_t _nvmlDeviceGetBBXTimeData_v1(nvmlDevice_t device, nvmlBBXTimeData_v1_t* timeData) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil
cdef nvmlReturn_t _nvmlDeviceGetAccountingStats_v2(nvmlDevice_t device, nvmlAccountingStats_v2_t* stats) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil
cdef nvmlReturn_t _nvmlDeviceGetRemappedRows_v2(nvmlDevice_t device, nvmlRemappedRowsInfo_v2_t* info) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil
86 changes: 85 additions & 1 deletion cuda_bindings/cuda/bindings/_internal/nvml_linux.pyx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
# SPDX-License-Identifier: Apache-2.0
#
# This code was automatically generated across versions from 12.9.1 to 13.3.0. Do not modify it directly.
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=646d902675de6987cff08e5702d3b2bff889e6a827273e57413aa5de45a5897a
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=45cd03eeb8717b33a1e11d83ae99ac8d6b580e4c57fad99d185614bd0047faf3


# <<<< PREAMBLE CONTENT >>>>
Expand Down Expand Up @@ -415,6 +415,10 @@ cdef void* __nvmlDeviceGetVgpuSchedulerLog_v2 = NULL
cdef void* __nvmlGpuInstanceGetVgpuSchedulerLog_v2 = NULL
cdef void* __nvmlDeviceSetVgpuSchedulerState_v2 = NULL
cdef void* __nvmlGpuInstanceSetVgpuSchedulerState_v2 = NULL
cdef void* __nvmlSystemGetCPER_v1 = NULL
cdef void* __nvmlDeviceGetBBXTimeData_v1 = NULL
cdef void* __nvmlDeviceGetAccountingStats_v2 = NULL
cdef void* __nvmlDeviceGetRemappedRows_v2 = NULL

cdef int _init_nvml() except -1 nogil:
global _cyb___py_nvml_init
Expand Down Expand Up @@ -2879,6 +2883,34 @@ cdef int _init_nvml() except -1 nogil:
handle = load_library()
__nvmlGpuInstanceSetVgpuSchedulerState_v2 = _cyb_dlsym(handle, 'nvmlGpuInstanceSetVgpuSchedulerState_v2')

global __nvmlSystemGetCPER_v1
__nvmlSystemGetCPER_v1 = _cyb_dlsym(_cyb_RTLD_DEFAULT, 'nvmlSystemGetCPER_v1')
if __nvmlSystemGetCPER_v1 == NULL:
if handle == NULL:
handle = load_library()
__nvmlSystemGetCPER_v1 = _cyb_dlsym(handle, 'nvmlSystemGetCPER_v1')

global __nvmlDeviceGetBBXTimeData_v1
__nvmlDeviceGetBBXTimeData_v1 = _cyb_dlsym(_cyb_RTLD_DEFAULT, 'nvmlDeviceGetBBXTimeData_v1')
if __nvmlDeviceGetBBXTimeData_v1 == NULL:
if handle == NULL:
handle = load_library()
__nvmlDeviceGetBBXTimeData_v1 = _cyb_dlsym(handle, 'nvmlDeviceGetBBXTimeData_v1')

global __nvmlDeviceGetAccountingStats_v2
__nvmlDeviceGetAccountingStats_v2 = _cyb_dlsym(_cyb_RTLD_DEFAULT, 'nvmlDeviceGetAccountingStats_v2')
if __nvmlDeviceGetAccountingStats_v2 == NULL:
if handle == NULL:
handle = load_library()
__nvmlDeviceGetAccountingStats_v2 = _cyb_dlsym(handle, 'nvmlDeviceGetAccountingStats_v2')

global __nvmlDeviceGetRemappedRows_v2
__nvmlDeviceGetRemappedRows_v2 = _cyb_dlsym(_cyb_RTLD_DEFAULT, 'nvmlDeviceGetRemappedRows_v2')
if __nvmlDeviceGetRemappedRows_v2 == NULL:
if handle == NULL:
handle = load_library()
__nvmlDeviceGetRemappedRows_v2 = _cyb_dlsym(handle, 'nvmlDeviceGetRemappedRows_v2')

_cyb_atomic_int_store(<int *>&_cyb___py_nvml_init, 1)
return 0

Expand Down Expand Up @@ -3948,6 +3980,18 @@ cpdef dict _inspect_function_pointers():

global __nvmlGpuInstanceSetVgpuSchedulerState_v2
data["__nvmlGpuInstanceSetVgpuSchedulerState_v2"] = <_cyb_intptr_t>__nvmlGpuInstanceSetVgpuSchedulerState_v2

global __nvmlSystemGetCPER_v1
data["__nvmlSystemGetCPER_v1"] = <_cyb_intptr_t>__nvmlSystemGetCPER_v1

global __nvmlDeviceGetBBXTimeData_v1
data["__nvmlDeviceGetBBXTimeData_v1"] = <_cyb_intptr_t>__nvmlDeviceGetBBXTimeData_v1

global __nvmlDeviceGetAccountingStats_v2
data["__nvmlDeviceGetAccountingStats_v2"] = <_cyb_intptr_t>__nvmlDeviceGetAccountingStats_v2

global __nvmlDeviceGetRemappedRows_v2
data["__nvmlDeviceGetRemappedRows_v2"] = <_cyb_intptr_t>__nvmlDeviceGetRemappedRows_v2
_cyb_func_ptrs = data
return data

Expand Down Expand Up @@ -7478,3 +7522,43 @@ cdef nvmlReturn_t _nvmlGpuInstanceSetVgpuSchedulerState_v2(nvmlGpuInstance_t gpu
raise FunctionNotFoundError("function nvmlGpuInstanceSetVgpuSchedulerState_v2 is not found")
return (<nvmlReturn_t (*)(nvmlGpuInstance_t, nvmlVgpuSchedulerState_v2_t*) noexcept nogil>__nvmlGpuInstanceSetVgpuSchedulerState_v2)(
gpuInstance, pSchedulerState)


cdef nvmlReturn_t _nvmlSystemGetCPER_v1(nvmlGetCPER_v1_t* cper) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil:
global __nvmlSystemGetCPER_v1
_check_or_init_nvml()
if __nvmlSystemGetCPER_v1 == NULL:
with gil:
raise FunctionNotFoundError("function nvmlSystemGetCPER_v1 is not found")
return (<nvmlReturn_t (*)(nvmlGetCPER_v1_t*) noexcept nogil>__nvmlSystemGetCPER_v1)(
cper)


cdef nvmlReturn_t _nvmlDeviceGetBBXTimeData_v1(nvmlDevice_t device, nvmlBBXTimeData_v1_t* timeData) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil:
global __nvmlDeviceGetBBXTimeData_v1
_check_or_init_nvml()
if __nvmlDeviceGetBBXTimeData_v1 == NULL:
with gil:
raise FunctionNotFoundError("function nvmlDeviceGetBBXTimeData_v1 is not found")
return (<nvmlReturn_t (*)(nvmlDevice_t, nvmlBBXTimeData_v1_t*) noexcept nogil>__nvmlDeviceGetBBXTimeData_v1)(
device, timeData)


cdef nvmlReturn_t _nvmlDeviceGetAccountingStats_v2(nvmlDevice_t device, nvmlAccountingStats_v2_t* stats) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil:
global __nvmlDeviceGetAccountingStats_v2
_check_or_init_nvml()
if __nvmlDeviceGetAccountingStats_v2 == NULL:
with gil:
raise FunctionNotFoundError("function nvmlDeviceGetAccountingStats_v2 is not found")
return (<nvmlReturn_t (*)(nvmlDevice_t, nvmlAccountingStats_v2_t*) noexcept nogil>__nvmlDeviceGetAccountingStats_v2)(
device, stats)


cdef nvmlReturn_t _nvmlDeviceGetRemappedRows_v2(nvmlDevice_t device, nvmlRemappedRowsInfo_v2_t* info) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil:
global __nvmlDeviceGetRemappedRows_v2
_check_or_init_nvml()
if __nvmlDeviceGetRemappedRows_v2 == NULL:
with gil:
raise FunctionNotFoundError("function nvmlDeviceGetRemappedRows_v2 is not found")
return (<nvmlReturn_t (*)(nvmlDevice_t, nvmlRemappedRowsInfo_v2_t*) noexcept nogil>__nvmlDeviceGetRemappedRows_v2)(
device, info)
70 changes: 69 additions & 1 deletion cuda_bindings/cuda/bindings/_internal/nvml_windows.pyx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
# SPDX-License-Identifier: Apache-2.0
#
# This code was automatically generated across versions from 12.9.1 to 13.3.0. Do not modify it directly.
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=831330186c4a7bb029b953be6dd3cda119a7f10c56fa9378ab621fbc4374f9d1
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=d7b5ba031ed135b60903431812f9883efdd554268990e04003d5ead5674466eb


# <<<< PREAMBLE CONTENT >>>>
Expand Down Expand Up @@ -412,6 +412,10 @@ cdef void* __nvmlDeviceGetVgpuSchedulerLog_v2 = NULL
cdef void* __nvmlGpuInstanceGetVgpuSchedulerLog_v2 = NULL
cdef void* __nvmlDeviceSetVgpuSchedulerState_v2 = NULL
cdef void* __nvmlGpuInstanceSetVgpuSchedulerState_v2 = NULL
cdef void* __nvmlSystemGetCPER_v1 = NULL
cdef void* __nvmlDeviceGetBBXTimeData_v1 = NULL
cdef void* __nvmlDeviceGetAccountingStats_v2 = NULL
cdef void* __nvmlDeviceGetRemappedRows_v2 = NULL

cdef int _init_nvml() except -1 nogil:
global _cyb___py_nvml_init
Expand Down Expand Up @@ -1475,6 +1479,18 @@ cdef int _init_nvml() except -1 nogil:
global __nvmlGpuInstanceSetVgpuSchedulerState_v2
__nvmlGpuInstanceSetVgpuSchedulerState_v2 = _cyb_GetProcAddress(<HMODULE>handle, 'nvmlGpuInstanceSetVgpuSchedulerState_v2')

global __nvmlSystemGetCPER_v1
__nvmlSystemGetCPER_v1 = _cyb_GetProcAddress(<HMODULE>handle, 'nvmlSystemGetCPER_v1')

global __nvmlDeviceGetBBXTimeData_v1
__nvmlDeviceGetBBXTimeData_v1 = _cyb_GetProcAddress(<HMODULE>handle, 'nvmlDeviceGetBBXTimeData_v1')

global __nvmlDeviceGetAccountingStats_v2
__nvmlDeviceGetAccountingStats_v2 = _cyb_GetProcAddress(<HMODULE>handle, 'nvmlDeviceGetAccountingStats_v2')

global __nvmlDeviceGetRemappedRows_v2
__nvmlDeviceGetRemappedRows_v2 = _cyb_GetProcAddress(<HMODULE>handle, 'nvmlDeviceGetRemappedRows_v2')

_cyb_atomic_int_store(<int *>&_cyb___py_nvml_init, 1)
return 0

Expand Down Expand Up @@ -2544,6 +2560,18 @@ cpdef dict _inspect_function_pointers():

global __nvmlGpuInstanceSetVgpuSchedulerState_v2
data["__nvmlGpuInstanceSetVgpuSchedulerState_v2"] = <_cyb_intptr_t>__nvmlGpuInstanceSetVgpuSchedulerState_v2

global __nvmlSystemGetCPER_v1
data["__nvmlSystemGetCPER_v1"] = <_cyb_intptr_t>__nvmlSystemGetCPER_v1

global __nvmlDeviceGetBBXTimeData_v1
data["__nvmlDeviceGetBBXTimeData_v1"] = <_cyb_intptr_t>__nvmlDeviceGetBBXTimeData_v1

global __nvmlDeviceGetAccountingStats_v2
data["__nvmlDeviceGetAccountingStats_v2"] = <_cyb_intptr_t>__nvmlDeviceGetAccountingStats_v2

global __nvmlDeviceGetRemappedRows_v2
data["__nvmlDeviceGetRemappedRows_v2"] = <_cyb_intptr_t>__nvmlDeviceGetRemappedRows_v2
_cyb_func_ptrs = data
return data

Expand Down Expand Up @@ -6073,3 +6101,43 @@ cdef nvmlReturn_t _nvmlGpuInstanceSetVgpuSchedulerState_v2(nvmlGpuInstance_t gpu
raise FunctionNotFoundError("function nvmlGpuInstanceSetVgpuSchedulerState_v2 is not found")
return (<nvmlReturn_t (*)(nvmlGpuInstance_t, nvmlVgpuSchedulerState_v2_t*) noexcept nogil>__nvmlGpuInstanceSetVgpuSchedulerState_v2)(
gpuInstance, pSchedulerState)


cdef nvmlReturn_t _nvmlSystemGetCPER_v1(nvmlGetCPER_v1_t* cper) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil:
global __nvmlSystemGetCPER_v1
_check_or_init_nvml()
if __nvmlSystemGetCPER_v1 == NULL:
with gil:
raise FunctionNotFoundError("function nvmlSystemGetCPER_v1 is not found")
return (<nvmlReturn_t (*)(nvmlGetCPER_v1_t*) noexcept nogil>__nvmlSystemGetCPER_v1)(
cper)


cdef nvmlReturn_t _nvmlDeviceGetBBXTimeData_v1(nvmlDevice_t device, nvmlBBXTimeData_v1_t* timeData) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil:
global __nvmlDeviceGetBBXTimeData_v1
_check_or_init_nvml()
if __nvmlDeviceGetBBXTimeData_v1 == NULL:
with gil:
raise FunctionNotFoundError("function nvmlDeviceGetBBXTimeData_v1 is not found")
return (<nvmlReturn_t (*)(nvmlDevice_t, nvmlBBXTimeData_v1_t*) noexcept nogil>__nvmlDeviceGetBBXTimeData_v1)(
device, timeData)


cdef nvmlReturn_t _nvmlDeviceGetAccountingStats_v2(nvmlDevice_t device, nvmlAccountingStats_v2_t* stats) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil:
global __nvmlDeviceGetAccountingStats_v2
_check_or_init_nvml()
if __nvmlDeviceGetAccountingStats_v2 == NULL:
with gil:
raise FunctionNotFoundError("function nvmlDeviceGetAccountingStats_v2 is not found")
return (<nvmlReturn_t (*)(nvmlDevice_t, nvmlAccountingStats_v2_t*) noexcept nogil>__nvmlDeviceGetAccountingStats_v2)(
device, stats)


cdef nvmlReturn_t _nvmlDeviceGetRemappedRows_v2(nvmlDevice_t device, nvmlRemappedRowsInfo_v2_t* info) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil:
global __nvmlDeviceGetRemappedRows_v2
_check_or_init_nvml()
if __nvmlDeviceGetRemappedRows_v2 == NULL:
with gil:
raise FunctionNotFoundError("function nvmlDeviceGetRemappedRows_v2 is not found")
return (<nvmlReturn_t (*)(nvmlDevice_t, nvmlRemappedRowsInfo_v2_t*) noexcept nogil>__nvmlDeviceGetRemappedRows_v2)(
device, info)
46 changes: 31 additions & 15 deletions cuda_bindings/cuda/bindings/cufile.pyx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
# SPDX-License-Identifier: Apache-2.0
#
# This code was automatically generated across versions from 12.9.1 to 13.3.0. Do not modify it directly.
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=4567ca64d02631fc8d6ed11af8164d72c86b4686c105fd733d186e6c92512749
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=9d2dd7c2d58ea1571aa5e0c4b88a498247727395a9d6864e66abef567efa41ff


# <<<< PREAMBLE CONTENT >>>>
Expand Down Expand Up @@ -2876,10 +2876,12 @@ cpdef intptr_t handle_register(intptr_t descr) except? 0:
"""cuFileHandleRegister is required, and performs extra checking that is memoized to provide increased performance on later cuFile operations.

Args:
descr (intptr_t): ``CUfileDescr_t`` file descriptor (OS agnostic).
descr (intptr_t): ``CUfileDescr_t`` file descriptor (OS
agnostic).

Returns:
intptr_t: ``CUfileHandle_t`` opaque file handle for IO operations.
intptr_t: ``CUfileHandle_t`` opaque file handle for IO
operations.

.. seealso:: `cuFileHandleRegister`
"""
Expand Down Expand Up @@ -2907,7 +2909,8 @@ cpdef buf_register(intptr_t buf_ptr_base, size_t length, int flags):

Args:
buf_ptr_base (intptr_t): buffer pointer allocated.
length (size_t): size of memory region from the above specified bufPtr.
length (size_t): size of memory region from the above
specified bufPtr.
flags (int): CU_FILE_RDMA_REGISTER.

.. seealso:: `cuFileBufRegister`
Expand Down Expand Up @@ -2967,8 +2970,10 @@ cpdef driver_set_poll_mode(bint poll, size_t poll_threshold_size):
"""Sets whether the Read/Write APIs use polling to do IO operations This takes place before the driver is opened. No-op if driver is already open.

Args:
poll (bint): boolean to indicate whether to use poll mode or not.
poll_threshold_size (size_t): max IO size to use for POLLING mode in KB.
poll (bint): boolean to indicate whether to use poll mode or
not.
poll_threshold_size (size_t): max IO size to use for POLLING
mode in KB.

.. seealso:: `cuFileDriverSetPollMode`
"""
Expand All @@ -2981,7 +2986,8 @@ cpdef driver_set_max_direct_io_size(size_t max_direct_io_size):
"""Control parameter to set max IO size(KB) used by the library to talk to nvidia-fs driver This takes place before the driver is opened. No-op if driver is already open.

Args:
max_direct_io_size (size_t): maximum allowed direct io size in KB.
max_direct_io_size (size_t): maximum allowed direct io size in
KB.

.. seealso:: `cuFileDriverSetMaxDirectIOSize`
"""
Expand All @@ -2994,7 +3000,8 @@ cpdef driver_set_max_cache_size(size_t max_cache_size):
"""Control parameter to set maximum GPU memory reserved per device by the library for internal buffering This takes place before the driver is opened. No-op if driver is already open.

Args:
max_cache_size (size_t): The maximum GPU buffer space per device used for internal use in KB.
max_cache_size (size_t): The maximum GPU buffer space per
device used for internal use in KB.

.. seealso:: `cuFileDriverSetMaxCacheSize`
"""
Expand All @@ -3007,7 +3014,8 @@ cpdef driver_set_max_pinned_mem_size(size_t max_pinned_size):
"""Sets maximum buffer space that is pinned in KB for use by ``cuFileBufRegister`` This takes place before the driver is opened. No-op if driver is already open.

Args:
max_pinned_size (size_t): maximum buffer space that is pinned in KB.
max_pinned_size (size_t): maximum buffer space that is pinned
in KB.

.. seealso:: `cuFileDriverSetMaxPinnedMemSize`
"""
Expand Down Expand Up @@ -3133,7 +3141,8 @@ cpdef tuple get_parameter_min_max_value(int param):
"""Get both the minimum and maximum settable values for a given size_t parameter in a single call.

Args:
param (SizeTConfigParameter): CUfile SizeT configuration parameter.
param (SizeTConfigParameter): CUfile SizeT configuration
parameter.

Returns:
A 2-tuple containing:
Expand All @@ -3155,7 +3164,8 @@ cpdef set_stats_level(int level):
"""Set the level of statistics collection for cuFile operations. This will override the cufile.json settings for stats.

Args:
level (int): Statistics level (0 = disabled, 1 = basic, 2 = detailed, 3 = verbose).
level (int): Statistics level (0 = disabled, 1 = basic, 2 =
detailed, 3 = verbose).

.. seealso:: `cuFileSetStatsLevel`
"""
Expand Down Expand Up @@ -3213,7 +3223,8 @@ cpdef get_stats_l1(intptr_t stats):
"""Get Level 1 cuFile statistics.

Args:
stats (intptr_t): Pointer to ``CUfileStatsLevel1_t`` structure to be filled.
stats (intptr_t): Pointer to ``CUfileStatsLevel1_t`` structure
to be filled.

.. seealso:: `cuFileGetStatsL1`
"""
Expand All @@ -3226,7 +3237,8 @@ cpdef get_stats_l2(intptr_t stats):
"""Get Level 2 cuFile statistics.

Args:
stats (intptr_t): Pointer to ``CUfileStatsLevel2_t`` structure to be filled.
stats (intptr_t): Pointer to ``CUfileStatsLevel2_t`` structure
to be filled.

.. seealso:: `cuFileGetStatsL2`
"""
Expand All @@ -3239,7 +3251,8 @@ cpdef get_stats_l3(intptr_t stats):
"""Get Level 3 cuFile statistics.

Args:
stats (intptr_t): Pointer to ``CUfileStatsLevel3_t`` structure to be filled.
stats (intptr_t): Pointer to ``CUfileStatsLevel3_t`` structure
to be filled.

.. seealso:: `cuFileGetStatsL3`
"""
Expand Down Expand Up @@ -3277,7 +3290,8 @@ cpdef get_parameter_posix_pool_slab_array(intptr_t size_values, intptr_t count_v
Args:
size_values (intptr_t): Buffer to receive slab sizes in KB.
count_values (intptr_t): Buffer to receive slab counts.
len (int): Buffer size (must match the actual parameter length).
len (int): Buffer size (must match the actual parameter
length).

.. seealso:: `cuFileGetParameterPosixPoolSlabArray`
"""
Expand Down Expand Up @@ -3346,4 +3360,6 @@ cpdef write(intptr_t fh, intptr_t buf_ptr_base, size_t size, off_t file_offset,
status = cuFileWrite(<Handle>fh, <const void*>buf_ptr_base, size, file_offset, buf_ptr_offset)
check_status(status)
return status


del _cyb_FastEnum
3 changes: 2 additions & 1 deletion cuda_bindings/cuda/bindings/cydriver.pxd
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
#
# This code was automatically generated across versions from 12.9.0 to 13.3.0. Do not modify it directly.

# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=cccb0572002cd20232f2b9f5c7acf559c92813d33dfc364136d57c8f453e50c6
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=3d716616a92d8ac919b59eccac24b84cb45d655dfb75436b7f9714c71d6f39e6
from libc.stdint cimport uint32_t, uint64_t


Expand Down Expand Up @@ -3716,6 +3716,7 @@ cdef enum: CU_MEM_CREATE_USAGE_TILE_POOL = 1

cdef enum: CU_MEM_CREATE_USAGE_HW_DECOMPRESS = 2


cdef enum: CU_MEM_POOL_CREATE_USAGE_HW_DECOMPRESS = 2

cdef enum: CUDA_COOPERATIVE_LAUNCH_MULTI_DEVICE_NO_PRE_LAUNCH_SYNC = 1
Expand Down
Loading
Loading