Skip to content
Draft
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
34 changes: 27 additions & 7 deletions cuda_bindings/cuda/bindings/_v2/nvrtc.pyx
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.0 to 13.3.0. Do not modify it directly.
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=a36c7e54cf29166832dd9aebc1fa71cc3649498794a2846e707396419caebe10
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=e48732452ebf6c1dda859086b349512cf0f2ef367fc16c46e887817a122d2e02


# <<<< PREAMBLE CONTENT >>>>
Expand Down Expand Up @@ -640,7 +640,6 @@ cpdef tuple version():

Returns:
A 2-tuple containing:

- int: CUDA Runtime Compilation major version number.
- int: CUDA Runtime Compilation minor version number.

Expand All @@ -657,6 +656,7 @@ cpdef tuple version():
cpdef int get_num_supported_archs() except? -1:
"""nvrtcGetNumSupportedArchs sets the output parameter ``num_archs`` with the number of architectures supported by NVRTC. This can then be used to pass an array to ``nvrtcGetSupportedArchs`` to get the supported architectures.

see ``nvrtcGetSupportedArchs``.
Returns:
int: number of supported architectures.

Expand All @@ -672,6 +672,7 @@ cpdef int get_num_supported_archs() except? -1:
cpdef object get_supported_archs():
"""nvrtcGetSupportedArchs populates the array passed via the output parameter ``supported_archs`` with the architectures supported by NVRTC. The array is sorted in the ascending order. The size of the array to be passed can be determined using ``nvrtcGetNumSupportedArchs``.

see ``nvrtcGetNumSupportedArchs``.
Returns:
int: sorted array of supported architectures.

Expand Down Expand Up @@ -881,6 +882,8 @@ cpdef bytes get_optix_ir(intptr_t prog):
cpdef size_t get_program_log_size(intptr_t prog) except? 0:
"""nvrtcGetProgramLogSize sets ``log_size_ret`` with the size of the log generated by the previous compilation of ``prog`` (including the trailing ``NULL``).

Note that compilation log may be generated with warnings and informative
messages, even when the compilation of ``prog`` succeeds.
Args:
prog (intptr_t): CUDA Runtime Compilation program.

Expand Down Expand Up @@ -925,11 +928,12 @@ cpdef bytes get_program_log(intptr_t prog):
cpdef add_name_expression(intptr_t prog, name_expression):
"""nvrtcAddNameExpression notes the given name expression denoting the address of a global function or device/__constant__ variable.

The identical name expression string must be provided on a subsequent call to
nvrtcGetLoweredName to extract the lowered name.
Args:
prog (intptr_t): CUDA Runtime Compilation program.
name_expression (str): constant expression denoting the
address of a global function or device/__constant__
variable.
name_expression (str): constant expression denoting the address of
a global function or device/__constant__ variable.

.. seealso:: `nvrtcAddNameExpression`
"""
Expand All @@ -946,8 +950,8 @@ cpdef size_t get_pch_heap_size() except? 0:
"""retrieve the current size of the PCH Heap.

Returns:
size_t: pointer to location where the size of the PCH Heap
will be stored.
size_t: pointer to location where the size of the PCH Heap will be
stored.

.. seealso:: `nvrtcGetPCHHeapSize`
"""
Expand All @@ -961,6 +965,9 @@ cpdef size_t get_pch_heap_size() except? 0:
cpdef set_pch_heap_size(size_t size):
"""set the size of the PCH Heap.

The requested size may be rounded up to a platform dependent alignment (e.g.
page size). If the PCH Heap has already been allocated, the heap memory will be
freed and a new PCH Heap will be allocated.
Args:
size (size_t): requested size of the PCH Heap, in bytes.

Expand All @@ -974,6 +981,19 @@ cpdef set_pch_heap_size(size_t size):
cpdef int get_pch_create_status(intptr_t prog) except? -1:
"""returns the PCH creation status.

NVRTC_SUCCESS indicates that the PCH was successfully created.
NVRTC_ERROR_NO_PCH_CREATE_ATTEMPTED indicates that no PCH creation was
attempted, either because PCH functionality was not requested during the
preceding nvrtcCompileProgram call, or automatic PCH processing was requested,
and compiler chose not to create a PCH file.
NVRTC_ERROR_PCH_CREATE_HEAP_EXHAUSTED indicates that a PCH file could
potentially have been created, but the compiler ran out space in the PCH heap.
In this scenario, the :func:`get_pch_heap_size_required` can be used to query
the required heap size, the heap can be reallocated for this size with
:func:`set_pch_heap_size` and PCH creation may be reattempted again invoking
:func:`compile_program` with a new NVRTC program instance.
NVRTC_ERROR_PCH_CREATE indicates that an error condition prevented the PCH file
from being created.
Args:
prog (intptr_t): CUDA Runtime Compilation program.

Expand Down
4 changes: 2 additions & 2 deletions cuda_bindings/cuda/bindings/cudla.pxd
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
# SPDX-License-Identifier: Apache-2.0

# This code was automatically generated across versions from 1.5.0 to 13.3.0. Do not modify it directly.
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=b82680ec867e23638b173760105c35030e0cba5c9a8b3bb536ce5bb3381ec1fb
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=3b52467aafbb78d33c99d8be6d3acdd552c8aa0851758d01932080aebaec5213


# <<<< PREAMBLE CONTENT >>>>
Expand Down Expand Up @@ -55,7 +55,7 @@ cpdef intptr_t create_device(uint64_t device, uint32_t flags) except *
cpdef intptr_t mem_register(intptr_t dev_handle, intptr_t ptr, size_t size, uint32_t flags) except *
cpdef intptr_t module_load_from_memory(intptr_t dev_handle, p_module, size_t module_size, uint32_t flags) except *
cpdef module_unload(intptr_t h_module, uint32_t flags)
cpdef submit_task(intptr_t dev_handle, intptr_t ptr_to_tasks, uint32_t num_tasks, intptr_t stream, uint32_t flags)
cpdef submit_task(intptr_t dev_handle, ptr_to_tasks, uint32_t num_tasks, intptr_t stream, uint32_t flags)
cpdef object device_get_attribute(intptr_t dev_handle, int attrib)
cpdef mem_unregister(intptr_t dev_handle, intptr_t dev_ptr)
cpdef int get_last_error(intptr_t dev_handle) except? 0
Expand Down
7 changes: 4 additions & 3 deletions cuda_bindings/cuda/bindings/cudla.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
# SPDX-License-Identifier: Apache-2.0

# This code was automatically generated across versions from 1.5.0 to 13.3.0. Do not modify it directly.
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=3c177b7a0328c0f6f16067c8c9f4e5a002bd019e8c17c017ba9f77af21da8d75
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=6448344915220edf6a2a1f4aee3377cee10c3c0a05db19c5ce37c6595bb52a5c


# <<<< PREAMBLE CONTENT >>>>
Expand Down Expand Up @@ -1804,9 +1804,10 @@ cpdef module_unload(intptr_t h_module, uint32_t flags):
check_status(__status__)


cpdef submit_task(intptr_t dev_handle, intptr_t ptr_to_tasks, uint32_t num_tasks, intptr_t stream, uint32_t flags):
cpdef submit_task(intptr_t dev_handle, ptr_to_tasks, uint32_t num_tasks, intptr_t stream, uint32_t flags):
cdef intptr_t _ptr_to_tasks_ptr_ = int(ptr_to_tasks)
with nogil:
__status__ = cudlaSubmitTask(<const DevHandle>dev_handle, <const cudlaTask* const>ptr_to_tasks, <const uint32_t>num_tasks, <void* const>stream, <const uint32_t>flags)
__status__ = cudlaSubmitTask(<const DevHandle>dev_handle, <const cudlaTask* const>_ptr_to_tasks_ptr_, <const uint32_t>num_tasks, <void* const>stream, <const uint32_t>flags)
check_status(__status__)


Expand Down
14 changes: 7 additions & 7 deletions cuda_bindings/cuda/bindings/cufile.pxd
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=b10e4f1751ee5423db23c6fc953cb0ae37bff7e8937bf1d39ac5fd6eeb0e4e87
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=232c1336341a3bd41bdcac568f24b320c839206e07a249bb4425ab726c1838be



Expand Down Expand Up @@ -53,7 +53,7 @@ ctypedef CUfileP2PFlags_t _P2PFlags
# Functions
###############################################################################

cpdef intptr_t handle_register(intptr_t descr) except? 0
cpdef intptr_t handle_register(descr) except? 0
cpdef void handle_deregister(intptr_t fh) except*
cpdef buf_register(intptr_t buf_ptr_base, size_t length, int flags)
cpdef buf_deregister(intptr_t buf_ptr_base)
Expand All @@ -65,8 +65,8 @@ cpdef driver_set_max_direct_io_size(size_t max_direct_io_size)
cpdef driver_set_max_cache_size(size_t max_cache_size)
cpdef driver_set_max_pinned_mem_size(size_t max_pinned_size)
cpdef intptr_t batch_io_set_up(unsigned nr) except? 0
cpdef batch_io_submit(intptr_t batch_idp, unsigned nr, intptr_t iocbp, unsigned int flags)
cpdef batch_io_get_status(intptr_t batch_idp, unsigned min_nr, intptr_t nr, intptr_t iocbp, intptr_t timeout)
cpdef batch_io_submit(intptr_t batch_idp, unsigned nr, iocbp, unsigned int flags)
cpdef batch_io_get_status(intptr_t batch_idp, unsigned min_nr, intptr_t nr, iocbp, intptr_t timeout)
cpdef batch_io_cancel(intptr_t batch_idp)
cpdef void batch_io_destroy(intptr_t batch_idp) except*
cpdef read_async(intptr_t fh, intptr_t buf_ptr_base, intptr_t size_p, intptr_t file_offset_p, intptr_t buf_ptr_offset_p, intptr_t bytes_read_p, intptr_t stream)
Expand All @@ -86,9 +86,9 @@ cpdef int get_stats_level() except? 0
cpdef stats_start()
cpdef stats_stop()
cpdef stats_reset()
cpdef get_stats_l1(intptr_t stats)
cpdef get_stats_l2(intptr_t stats)
cpdef get_stats_l3(intptr_t stats)
cpdef get_stats_l1(stats)
cpdef get_stats_l2(stats)
cpdef get_stats_l3(stats)
cpdef size_t get_bar_size_in_kb(int gpu_index) except? 0
cpdef set_parameter_posix_pool_slab_array(intptr_t size_values, intptr_t count_values, int len)
cpdef get_parameter_posix_pool_slab_array(intptr_t size_values, intptr_t count_values, int len)
Loading
Loading