Skip to content

Commit a9b679f

Browse files
committed
Transfer fix from codegen, NO manual changes.
1 parent 0afcf9e commit a9b679f

2 files changed

Lines changed: 21 additions & 0 deletions

File tree

cuda_bindings/cuda/bindings/cufile.pxd

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -74,3 +74,6 @@ cpdef int get_version() except? 0
7474
cpdef size_t get_parameter_size_t(int param) except? 0
7575
cpdef bint get_parameter_bool(int param) except? 0
7676
cpdef str get_parameter_string(int param, int len)
77+
cpdef set_parameter_size_t(int param, size_t value)
78+
cpdef set_parameter_bool(int param, bint value)
79+
cpdef set_parameter_string(int param, intptr_t desc_str)

cuda_bindings/cuda/bindings/cufile.pyx

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1273,6 +1273,24 @@ cpdef str get_parameter_string(int param, int len):
12731273
return _desc_str_.decode()
12741274

12751275

1276+
cpdef set_parameter_size_t(int param, size_t value):
1277+
with nogil:
1278+
status = cuFileSetParameterSizeT(<_SizeTConfigParameter>param, value)
1279+
check_status(status)
1280+
1281+
1282+
cpdef set_parameter_bool(int param, bint value):
1283+
with nogil:
1284+
status = cuFileSetParameterBool(<_BoolConfigParameter>param, <cpp_bool>value)
1285+
check_status(status)
1286+
1287+
1288+
cpdef set_parameter_string(int param, intptr_t desc_str):
1289+
with nogil:
1290+
status = cuFileSetParameterString(<_StringConfigParameter>param, <const char*>desc_str)
1291+
check_status(status)
1292+
1293+
12761294
cpdef str op_status_error(int status):
12771295
"""cufileop status string.
12781296

0 commit comments

Comments
 (0)