Skip to content

Commit 485531d

Browse files
rwgkrwgk
authored
Minimal cufile cuda_bindings updates for CTK 13 (#826)
* Transfer from codegen, NO manual changes. * Transfer fix from codegen, NO manual changes.
1 parent 2400e46 commit 485531d

6 files changed

Lines changed: 481 additions & 6 deletions

File tree

‎cuda_bindings/cuda/bindings/_internal/cufile.pxd‎

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
#
33
# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
44
#
5-
# This code was automatically generated with version 12.9.0. Do not modify it directly.
5+
# This code was automatically generated across versions from 12.9.0 to 13.0.0. Do not modify it directly.
66

77
from ..cycufile cimport *
88

@@ -41,3 +41,16 @@ cdef CUfileError_t _cuFileGetParameterString(CUFileStringConfigParameter_t param
4141
cdef CUfileError_t _cuFileSetParameterSizeT(CUFileSizeTConfigParameter_t param, size_t value) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
4242
cdef CUfileError_t _cuFileSetParameterBool(CUFileBoolConfigParameter_t param, cpp_bool value) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
4343
cdef CUfileError_t _cuFileSetParameterString(CUFileStringConfigParameter_t param, const char* desc_str) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
44+
cdef CUfileError_t _cuFileDriverClose() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
45+
cdef CUfileError_t _cuFileGetParameterMinMaxValue(CUFileSizeTConfigParameter_t param, size_t* min_value, size_t* max_value) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
46+
cdef CUfileError_t _cuFileSetStatsLevel(int level) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
47+
cdef CUfileError_t _cuFileGetStatsLevel(int* level) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
48+
cdef CUfileError_t _cuFileStatsStart() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
49+
cdef CUfileError_t _cuFileStatsStop() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
50+
cdef CUfileError_t _cuFileStatsReset() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
51+
cdef CUfileError_t _cuFileGetStatsL1(CUfileStatsLevel1_t* stats) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
52+
cdef CUfileError_t _cuFileGetStatsL2(CUfileStatsLevel2_t* stats) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
53+
cdef CUfileError_t _cuFileGetStatsL3(CUfileStatsLevel3_t* stats) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
54+
cdef CUfileError_t _cuFileGetBARSizeInKB(int gpuIndex, size_t* barSize) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
55+
cdef CUfileError_t _cuFileSetParameterPosixPoolSlabArray(const size_t* size_values, const size_t* count_values, int len) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil
56+
cdef CUfileError_t _cuFileGetParameterPosixPoolSlabArray(size_t* size_values, size_t* count_values, int len) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil

‎cuda_bindings/cuda/bindings/_internal/cufile_linux.pyx‎

Lines changed: 274 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
#
33
# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
44
#
5-
# This code was automatically generated with version 12.9.0. Do not modify it directly.
5+
# This code was automatically generated across versions from 12.9.0 to 13.0.0. Do not modify it directly.
66

77
from libc.stdint cimport intptr_t, uintptr_t
88

@@ -68,6 +68,19 @@ cdef void* __cuFileGetParameterString = NULL
6868
cdef void* __cuFileSetParameterSizeT = NULL
6969
cdef void* __cuFileSetParameterBool = NULL
7070
cdef void* __cuFileSetParameterString = NULL
71+
cdef void* __cuFileDriverClose = NULL
72+
cdef void* __cuFileGetParameterMinMaxValue = NULL
73+
cdef void* __cuFileSetStatsLevel = NULL
74+
cdef void* __cuFileGetStatsLevel = NULL
75+
cdef void* __cuFileStatsStart = NULL
76+
cdef void* __cuFileStatsStop = NULL
77+
cdef void* __cuFileStatsReset = NULL
78+
cdef void* __cuFileGetStatsL1 = NULL
79+
cdef void* __cuFileGetStatsL2 = NULL
80+
cdef void* __cuFileGetStatsL3 = NULL
81+
cdef void* __cuFileGetBARSizeInKB = NULL
82+
cdef void* __cuFileSetParameterPosixPoolSlabArray = NULL
83+
cdef void* __cuFileGetParameterPosixPoolSlabArray = NULL
7184

7285

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

328+
global __cuFileDriverClose
329+
__cuFileDriverClose = dlsym(RTLD_DEFAULT, 'cuFileDriverClose')
330+
if __cuFileDriverClose == NULL:
331+
if handle == NULL:
332+
handle = load_library(driver_ver)
333+
__cuFileDriverClose = dlsym(handle, 'cuFileDriverClose')
334+
335+
global __cuFileGetParameterMinMaxValue
336+
__cuFileGetParameterMinMaxValue = dlsym(RTLD_DEFAULT, 'cuFileGetParameterMinMaxValue')
337+
if __cuFileGetParameterMinMaxValue == NULL:
338+
if handle == NULL:
339+
handle = load_library(driver_ver)
340+
__cuFileGetParameterMinMaxValue = dlsym(handle, 'cuFileGetParameterMinMaxValue')
341+
342+
global __cuFileSetStatsLevel
343+
__cuFileSetStatsLevel = dlsym(RTLD_DEFAULT, 'cuFileSetStatsLevel')
344+
if __cuFileSetStatsLevel == NULL:
345+
if handle == NULL:
346+
handle = load_library(driver_ver)
347+
__cuFileSetStatsLevel = dlsym(handle, 'cuFileSetStatsLevel')
348+
349+
global __cuFileGetStatsLevel
350+
__cuFileGetStatsLevel = dlsym(RTLD_DEFAULT, 'cuFileGetStatsLevel')
351+
if __cuFileGetStatsLevel == NULL:
352+
if handle == NULL:
353+
handle = load_library(driver_ver)
354+
__cuFileGetStatsLevel = dlsym(handle, 'cuFileGetStatsLevel')
355+
356+
global __cuFileStatsStart
357+
__cuFileStatsStart = dlsym(RTLD_DEFAULT, 'cuFileStatsStart')
358+
if __cuFileStatsStart == NULL:
359+
if handle == NULL:
360+
handle = load_library(driver_ver)
361+
__cuFileStatsStart = dlsym(handle, 'cuFileStatsStart')
362+
363+
global __cuFileStatsStop
364+
__cuFileStatsStop = dlsym(RTLD_DEFAULT, 'cuFileStatsStop')
365+
if __cuFileStatsStop == NULL:
366+
if handle == NULL:
367+
handle = load_library(driver_ver)
368+
__cuFileStatsStop = dlsym(handle, 'cuFileStatsStop')
369+
370+
global __cuFileStatsReset
371+
__cuFileStatsReset = dlsym(RTLD_DEFAULT, 'cuFileStatsReset')
372+
if __cuFileStatsReset == NULL:
373+
if handle == NULL:
374+
handle = load_library(driver_ver)
375+
__cuFileStatsReset = dlsym(handle, 'cuFileStatsReset')
376+
377+
global __cuFileGetStatsL1
378+
__cuFileGetStatsL1 = dlsym(RTLD_DEFAULT, 'cuFileGetStatsL1')
379+
if __cuFileGetStatsL1 == NULL:
380+
if handle == NULL:
381+
handle = load_library(driver_ver)
382+
__cuFileGetStatsL1 = dlsym(handle, 'cuFileGetStatsL1')
383+
384+
global __cuFileGetStatsL2
385+
__cuFileGetStatsL2 = dlsym(RTLD_DEFAULT, 'cuFileGetStatsL2')
386+
if __cuFileGetStatsL2 == NULL:
387+
if handle == NULL:
388+
handle = load_library(driver_ver)
389+
__cuFileGetStatsL2 = dlsym(handle, 'cuFileGetStatsL2')
390+
391+
global __cuFileGetStatsL3
392+
__cuFileGetStatsL3 = dlsym(RTLD_DEFAULT, 'cuFileGetStatsL3')
393+
if __cuFileGetStatsL3 == NULL:
394+
if handle == NULL:
395+
handle = load_library(driver_ver)
396+
__cuFileGetStatsL3 = dlsym(handle, 'cuFileGetStatsL3')
397+
398+
global __cuFileGetBARSizeInKB
399+
__cuFileGetBARSizeInKB = dlsym(RTLD_DEFAULT, 'cuFileGetBARSizeInKB')
400+
if __cuFileGetBARSizeInKB == NULL:
401+
if handle == NULL:
402+
handle = load_library(driver_ver)
403+
__cuFileGetBARSizeInKB = dlsym(handle, 'cuFileGetBARSizeInKB')
404+
405+
global __cuFileSetParameterPosixPoolSlabArray
406+
__cuFileSetParameterPosixPoolSlabArray = dlsym(RTLD_DEFAULT, 'cuFileSetParameterPosixPoolSlabArray')
407+
if __cuFileSetParameterPosixPoolSlabArray == NULL:
408+
if handle == NULL:
409+
handle = load_library(driver_ver)
410+
__cuFileSetParameterPosixPoolSlabArray = dlsym(handle, 'cuFileSetParameterPosixPoolSlabArray')
411+
412+
global __cuFileGetParameterPosixPoolSlabArray
413+
__cuFileGetParameterPosixPoolSlabArray = dlsym(RTLD_DEFAULT, 'cuFileGetParameterPosixPoolSlabArray')
414+
if __cuFileGetParameterPosixPoolSlabArray == NULL:
415+
if handle == NULL:
416+
handle = load_library(driver_ver)
417+
__cuFileGetParameterPosixPoolSlabArray = dlsym(handle, 'cuFileGetParameterPosixPoolSlabArray')
418+
315419
__py_cufile_init = True
316420
return 0
317421

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

524+
global __cuFileDriverClose
525+
data["__cuFileDriverClose"] = <intptr_t>__cuFileDriverClose
526+
527+
global __cuFileGetParameterMinMaxValue
528+
data["__cuFileGetParameterMinMaxValue"] = <intptr_t>__cuFileGetParameterMinMaxValue
529+
530+
global __cuFileSetStatsLevel
531+
data["__cuFileSetStatsLevel"] = <intptr_t>__cuFileSetStatsLevel
532+
533+
global __cuFileGetStatsLevel
534+
data["__cuFileGetStatsLevel"] = <intptr_t>__cuFileGetStatsLevel
535+
536+
global __cuFileStatsStart
537+
data["__cuFileStatsStart"] = <intptr_t>__cuFileStatsStart
538+
539+
global __cuFileStatsStop
540+
data["__cuFileStatsStop"] = <intptr_t>__cuFileStatsStop
541+
542+
global __cuFileStatsReset
543+
data["__cuFileStatsReset"] = <intptr_t>__cuFileStatsReset
544+
545+
global __cuFileGetStatsL1
546+
data["__cuFileGetStatsL1"] = <intptr_t>__cuFileGetStatsL1
547+
548+
global __cuFileGetStatsL2
549+
data["__cuFileGetStatsL2"] = <intptr_t>__cuFileGetStatsL2
550+
551+
global __cuFileGetStatsL3
552+
data["__cuFileGetStatsL3"] = <intptr_t>__cuFileGetStatsL3
553+
554+
global __cuFileGetBARSizeInKB
555+
data["__cuFileGetBARSizeInKB"] = <intptr_t>__cuFileGetBARSizeInKB
556+
557+
global __cuFileSetParameterPosixPoolSlabArray
558+
data["__cuFileSetParameterPosixPoolSlabArray"] = <intptr_t>__cuFileSetParameterPosixPoolSlabArray
559+
560+
global __cuFileGetParameterPosixPoolSlabArray
561+
data["__cuFileGetParameterPosixPoolSlabArray"] = <intptr_t>__cuFileGetParameterPosixPoolSlabArray
562+
420563
func_ptrs = data
421564
return data
422565

@@ -732,3 +875,133 @@ cdef CUfileError_t _cuFileSetParameterString(CUFileStringConfigParameter_t param
732875
raise FunctionNotFoundError("function cuFileSetParameterString is not found")
733876
return (<CUfileError_t (*)(CUFileStringConfigParameter_t, const char*) noexcept nogil>__cuFileSetParameterString)(
734877
param, desc_str)
878+
879+
880+
cdef CUfileError_t _cuFileDriverClose() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
881+
global __cuFileDriverClose
882+
_check_or_init_cufile()
883+
if __cuFileDriverClose == NULL:
884+
with gil:
885+
raise FunctionNotFoundError("function cuFileDriverClose is not found")
886+
return (<CUfileError_t (*)() noexcept nogil>__cuFileDriverClose)(
887+
)
888+
889+
890+
cdef CUfileError_t _cuFileGetParameterMinMaxValue(CUFileSizeTConfigParameter_t param, size_t* min_value, size_t* max_value) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
891+
global __cuFileGetParameterMinMaxValue
892+
_check_or_init_cufile()
893+
if __cuFileGetParameterMinMaxValue == NULL:
894+
with gil:
895+
raise FunctionNotFoundError("function cuFileGetParameterMinMaxValue is not found")
896+
return (<CUfileError_t (*)(CUFileSizeTConfigParameter_t, size_t*, size_t*) noexcept nogil>__cuFileGetParameterMinMaxValue)(
897+
param, min_value, max_value)
898+
899+
900+
cdef CUfileError_t _cuFileSetStatsLevel(int level) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
901+
global __cuFileSetStatsLevel
902+
_check_or_init_cufile()
903+
if __cuFileSetStatsLevel == NULL:
904+
with gil:
905+
raise FunctionNotFoundError("function cuFileSetStatsLevel is not found")
906+
return (<CUfileError_t (*)(int) noexcept nogil>__cuFileSetStatsLevel)(
907+
level)
908+
909+
910+
cdef CUfileError_t _cuFileGetStatsLevel(int* level) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
911+
global __cuFileGetStatsLevel
912+
_check_or_init_cufile()
913+
if __cuFileGetStatsLevel == NULL:
914+
with gil:
915+
raise FunctionNotFoundError("function cuFileGetStatsLevel is not found")
916+
return (<CUfileError_t (*)(int*) noexcept nogil>__cuFileGetStatsLevel)(
917+
level)
918+
919+
920+
cdef CUfileError_t _cuFileStatsStart() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
921+
global __cuFileStatsStart
922+
_check_or_init_cufile()
923+
if __cuFileStatsStart == NULL:
924+
with gil:
925+
raise FunctionNotFoundError("function cuFileStatsStart is not found")
926+
return (<CUfileError_t (*)() noexcept nogil>__cuFileStatsStart)(
927+
)
928+
929+
930+
cdef CUfileError_t _cuFileStatsStop() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
931+
global __cuFileStatsStop
932+
_check_or_init_cufile()
933+
if __cuFileStatsStop == NULL:
934+
with gil:
935+
raise FunctionNotFoundError("function cuFileStatsStop is not found")
936+
return (<CUfileError_t (*)() noexcept nogil>__cuFileStatsStop)(
937+
)
938+
939+
940+
cdef CUfileError_t _cuFileStatsReset() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
941+
global __cuFileStatsReset
942+
_check_or_init_cufile()
943+
if __cuFileStatsReset == NULL:
944+
with gil:
945+
raise FunctionNotFoundError("function cuFileStatsReset is not found")
946+
return (<CUfileError_t (*)() noexcept nogil>__cuFileStatsReset)(
947+
)
948+
949+
950+
cdef CUfileError_t _cuFileGetStatsL1(CUfileStatsLevel1_t* stats) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
951+
global __cuFileGetStatsL1
952+
_check_or_init_cufile()
953+
if __cuFileGetStatsL1 == NULL:
954+
with gil:
955+
raise FunctionNotFoundError("function cuFileGetStatsL1 is not found")
956+
return (<CUfileError_t (*)(CUfileStatsLevel1_t*) noexcept nogil>__cuFileGetStatsL1)(
957+
stats)
958+
959+
960+
cdef CUfileError_t _cuFileGetStatsL2(CUfileStatsLevel2_t* stats) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
961+
global __cuFileGetStatsL2
962+
_check_or_init_cufile()
963+
if __cuFileGetStatsL2 == NULL:
964+
with gil:
965+
raise FunctionNotFoundError("function cuFileGetStatsL2 is not found")
966+
return (<CUfileError_t (*)(CUfileStatsLevel2_t*) noexcept nogil>__cuFileGetStatsL2)(
967+
stats)
968+
969+
970+
cdef CUfileError_t _cuFileGetStatsL3(CUfileStatsLevel3_t* stats) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
971+
global __cuFileGetStatsL3
972+
_check_or_init_cufile()
973+
if __cuFileGetStatsL3 == NULL:
974+
with gil:
975+
raise FunctionNotFoundError("function cuFileGetStatsL3 is not found")
976+
return (<CUfileError_t (*)(CUfileStatsLevel3_t*) noexcept nogil>__cuFileGetStatsL3)(
977+
stats)
978+
979+
980+
cdef CUfileError_t _cuFileGetBARSizeInKB(int gpuIndex, size_t* barSize) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
981+
global __cuFileGetBARSizeInKB
982+
_check_or_init_cufile()
983+
if __cuFileGetBARSizeInKB == NULL:
984+
with gil:
985+
raise FunctionNotFoundError("function cuFileGetBARSizeInKB is not found")
986+
return (<CUfileError_t (*)(int, size_t*) noexcept nogil>__cuFileGetBARSizeInKB)(
987+
gpuIndex, barSize)
988+
989+
990+
cdef CUfileError_t _cuFileSetParameterPosixPoolSlabArray(const size_t* size_values, const size_t* count_values, int len) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
991+
global __cuFileSetParameterPosixPoolSlabArray
992+
_check_or_init_cufile()
993+
if __cuFileSetParameterPosixPoolSlabArray == NULL:
994+
with gil:
995+
raise FunctionNotFoundError("function cuFileSetParameterPosixPoolSlabArray is not found")
996+
return (<CUfileError_t (*)(const size_t*, const size_t*, int) noexcept nogil>__cuFileSetParameterPosixPoolSlabArray)(
997+
size_values, count_values, len)
998+
999+
1000+
cdef CUfileError_t _cuFileGetParameterPosixPoolSlabArray(size_t* size_values, size_t* count_values, int len) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil:
1001+
global __cuFileGetParameterPosixPoolSlabArray
1002+
_check_or_init_cufile()
1003+
if __cuFileGetParameterPosixPoolSlabArray == NULL:
1004+
with gil:
1005+
raise FunctionNotFoundError("function cuFileGetParameterPosixPoolSlabArray is not found")
1006+
return (<CUfileError_t (*)(size_t*, size_t*, int) noexcept nogil>__cuFileGetParameterPosixPoolSlabArray)(
1007+
size_values, count_values, len)

‎cuda_bindings/cuda/bindings/cufile.pxd‎

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
#
33
# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
44
#
5-
# This code was automatically generated with version 12.9.0. Do not modify it directly.
5+
# This code was automatically generated across versions from 12.9.0 to 13.0.0. Do not modify it directly.
66

77
from libc.stdint cimport intptr_t
88

@@ -18,7 +18,12 @@ ctypedef CUfileBatchHandle_t BatchHandle
1818
ctypedef CUfileError_t Error
1919
ctypedef cufileRDMAInfo_t RDMAInfo
2020
ctypedef CUfileFSOps_t FSOps
21+
ctypedef CUfileOpCounter_t OpCounter
22+
ctypedef CUfilePerGpuStats_t PerGpuStats
2123
ctypedef CUfileDrvProps_t DrvProps
24+
ctypedef CUfileStatsLevel1_t StatsLevel1
25+
ctypedef CUfileStatsLevel2_t StatsLevel2
26+
ctypedef CUfileStatsLevel3_t StatsLevel3
2227

2328

2429
###############################################################################
@@ -36,6 +41,7 @@ ctypedef CUfileBatchMode_t _BatchMode
3641
ctypedef CUFileSizeTConfigParameter_t _SizeTConfigParameter
3742
ctypedef CUFileBoolConfigParameter_t _BoolConfigParameter
3843
ctypedef CUFileStringConfigParameter_t _StringConfigParameter
44+
ctypedef CUFileArrayConfigParameter_t _ArrayConfigParameter
3945

4046

4147
###############################################################################

0 commit comments

Comments
 (0)