Coverage for cuda / bindings / cycufile.pyx: 75%

1# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. 

2# 

3# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE 

4# 

5# This code was automatically generated across versions from 12.9.1 to 13.1.0. Do not modify it directly. 

6  

7from ._internal cimport cufile as _cufile 

8  

9import cython 

10  

11############################################################################### 

12# Wrapper functions 

13############################################################################### 

14  

15cdef CUfileError_t cuFileHandleRegister(CUfileHandle_t* fh, CUfileDescr_t* descr) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

16 return _cufile._cuFileHandleRegister(fh, descr) 

17  

18  

19@cython.show_performance_hints(False) 

20cdef void cuFileHandleDeregister(CUfileHandle_t fh) except* nogil: 

21 _cufile._cuFileHandleDeregister(fh) 

22  

23  

24cdef CUfileError_t cuFileBufRegister(const void* bufPtr_base, size_t length, int flags) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

25 return _cufile._cuFileBufRegister(bufPtr_base, length, flags) 

26  

27  

28cdef CUfileError_t cuFileBufDeregister(const void* bufPtr_base) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

29 return _cufile._cuFileBufDeregister(bufPtr_base) 

30  

31  

32cdef ssize_t cuFileRead(CUfileHandle_t fh, void* bufPtr_base, size_t size, off_t file_offset, off_t bufPtr_offset) except* nogil: 

33 return _cufile._cuFileRead(fh, bufPtr_base, size, file_offset, bufPtr_offset) 

34  

35  

36cdef ssize_t cuFileWrite(CUfileHandle_t fh, const void* bufPtr_base, size_t size, off_t file_offset, off_t bufPtr_offset) except* nogil: 

37 return _cufile._cuFileWrite(fh, bufPtr_base, size, file_offset, bufPtr_offset) 

38  

39  

40cdef CUfileError_t cuFileDriverOpen() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

41 return _cufile._cuFileDriverOpen() 

42  

43  

44cdef CUfileError_t cuFileDriverClose() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

45 return _cufile._cuFileDriverClose() 

46  

47  

48cdef CUfileError_t cuFileDriverClose_v2() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

49 return _cufile._cuFileDriverClose_v2() 

50  

51  

52cdef long cuFileUseCount() except* nogil: 

53 return _cufile._cuFileUseCount() 

54  

55  

56cdef CUfileError_t cuFileDriverGetProperties(CUfileDrvProps_t* props) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

57 return _cufile._cuFileDriverGetProperties(props) 

58  

59  

60cdef CUfileError_t cuFileDriverSetPollMode(cpp_bool poll, size_t poll_threshold_size) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

61 return _cufile._cuFileDriverSetPollMode(poll, poll_threshold_size) 

62  

63  

64cdef CUfileError_t cuFileDriverSetMaxDirectIOSize(size_t max_direct_io_size) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

65 return _cufile._cuFileDriverSetMaxDirectIOSize(max_direct_io_size) 

66  

67  

68cdef CUfileError_t cuFileDriverSetMaxCacheSize(size_t max_cache_size) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

69 return _cufile._cuFileDriverSetMaxCacheSize(max_cache_size) 

70  

71  

72cdef CUfileError_t cuFileDriverSetMaxPinnedMemSize(size_t max_pinned_size) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

73 return _cufile._cuFileDriverSetMaxPinnedMemSize(max_pinned_size) 

74  

75  

76cdef CUfileError_t cuFileBatchIOSetUp(CUfileBatchHandle_t* batch_idp, unsigned nr) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

77 return _cufile._cuFileBatchIOSetUp(batch_idp, nr) 

78  

79  

80cdef CUfileError_t cuFileBatchIOSubmit(CUfileBatchHandle_t batch_idp, unsigned nr, CUfileIOParams_t* iocbp, unsigned int flags) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

81 return _cufile._cuFileBatchIOSubmit(batch_idp, nr, iocbp, flags) 

82  

83  

84cdef CUfileError_t cuFileBatchIOGetStatus(CUfileBatchHandle_t batch_idp, unsigned min_nr, unsigned* nr, CUfileIOEvents_t* iocbp, timespec* timeout) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

85 return _cufile._cuFileBatchIOGetStatus(batch_idp, min_nr, nr, iocbp, timeout) 

86  

87  

88cdef CUfileError_t cuFileBatchIOCancel(CUfileBatchHandle_t batch_idp) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

89 return _cufile._cuFileBatchIOCancel(batch_idp) 

90  

91  

92@cython.show_performance_hints(False) 

93cdef void cuFileBatchIODestroy(CUfileBatchHandle_t batch_idp) except* nogil: 

94 _cufile._cuFileBatchIODestroy(batch_idp) 

95  

96  

97cdef CUfileError_t cuFileReadAsync(CUfileHandle_t fh, void* bufPtr_base, size_t* size_p, off_t* file_offset_p, off_t* bufPtr_offset_p, ssize_t* bytes_read_p, CUstream stream) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

98 return _cufile._cuFileReadAsync(fh, bufPtr_base, size_p, file_offset_p, bufPtr_offset_p, bytes_read_p, stream) 

99  

100  

101cdef CUfileError_t cuFileWriteAsync(CUfileHandle_t fh, void* bufPtr_base, size_t* size_p, off_t* file_offset_p, off_t* bufPtr_offset_p, ssize_t* bytes_written_p, CUstream stream) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

102 return _cufile._cuFileWriteAsync(fh, bufPtr_base, size_p, file_offset_p, bufPtr_offset_p, bytes_written_p, stream) 

103  

104  

105cdef CUfileError_t cuFileStreamRegister(CUstream stream, unsigned flags) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

106 return _cufile._cuFileStreamRegister(stream, flags) 

107  

108  

109cdef CUfileError_t cuFileStreamDeregister(CUstream stream) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

110 return _cufile._cuFileStreamDeregister(stream) 

111  

112  

113cdef CUfileError_t cuFileGetVersion(int* version) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

114 return _cufile._cuFileGetVersion(version) 

115  

116  

117cdef CUfileError_t cuFileGetParameterSizeT(CUFileSizeTConfigParameter_t param, size_t* value) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

118 return _cufile._cuFileGetParameterSizeT(param, value) 

119  

120  

121cdef CUfileError_t cuFileGetParameterBool(CUFileBoolConfigParameter_t param, cpp_bool* value) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

122 return _cufile._cuFileGetParameterBool(param, value) 

123  

124  

125cdef CUfileError_t cuFileGetParameterString(CUFileStringConfigParameter_t param, char* desc_str, int len) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

126 return _cufile._cuFileGetParameterString(param, desc_str, len) 

127  

128  

129cdef CUfileError_t cuFileSetParameterSizeT(CUFileSizeTConfigParameter_t param, size_t value) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

130 return _cufile._cuFileSetParameterSizeT(param, value) 

131  

132  

133cdef CUfileError_t cuFileSetParameterBool(CUFileBoolConfigParameter_t param, cpp_bool value) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

134 return _cufile._cuFileSetParameterBool(param, value) 

135  

136  

137cdef CUfileError_t cuFileSetParameterString(CUFileStringConfigParameter_t param, const char* desc_str) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

138 return _cufile._cuFileSetParameterString(param, desc_str) 

139  

140  

141cdef CUfileError_t cuFileGetParameterMinMaxValue(CUFileSizeTConfigParameter_t param, size_t* min_value, size_t* max_value) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

142 return _cufile._cuFileGetParameterMinMaxValue(param, min_value, max_value) 

143  

144  

145cdef CUfileError_t cuFileSetStatsLevel(int level) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

146 return _cufile._cuFileSetStatsLevel(level) 

147  

148  

149cdef CUfileError_t cuFileGetStatsLevel(int* level) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

150 return _cufile._cuFileGetStatsLevel(level) 

151  

152  

153cdef CUfileError_t cuFileStatsStart() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

154 return _cufile._cuFileStatsStart() 

155  

156  

157cdef CUfileError_t cuFileStatsStop() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

158 return _cufile._cuFileStatsStop() 

159  

160  

161cdef CUfileError_t cuFileStatsReset() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

162 return _cufile._cuFileStatsReset() 

163  

164  

165cdef CUfileError_t cuFileGetStatsL1(CUfileStatsLevel1_t* stats) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

166 return _cufile._cuFileGetStatsL1(stats) 

167  

168  

169cdef CUfileError_t cuFileGetStatsL2(CUfileStatsLevel2_t* stats) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

170 return _cufile._cuFileGetStatsL2(stats) 

171  

172  

173cdef CUfileError_t cuFileGetStatsL3(CUfileStatsLevel3_t* stats) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

174 return _cufile._cuFileGetStatsL3(stats) 

175  

176  

177cdef CUfileError_t cuFileGetBARSizeInKB(int gpuIndex, size_t* barSize) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

178 return _cufile._cuFileGetBARSizeInKB(gpuIndex, barSize) 

179  

180  

181cdef CUfileError_t cuFileSetParameterPosixPoolSlabArray(const size_t* size_values, const size_t* count_values, int len) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

182 return _cufile._cuFileSetParameterPosixPoolSlabArray(size_values, count_values, len) 

183  

184  

185cdef CUfileError_t cuFileGetParameterPosixPoolSlabArray(size_t* size_values, size_t* count_values, int len) except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: 

186 return _cufile._cuFileGetParameterPosixPoolSlabArray(size_values, count_values, len)