Coverage for cuda/bindings/cycudla.pyx: 51.85%

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

2# SPDX-License-Identifier: Apache-2.0 

3  

4# This code was automatically generated across versions from 1.5.0 to 13.3.0. Do not modify it directly. 

5# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=9e8c534ff8b9d4e348af657e66d81b4758d90e7a3b840705267282c5cc4e8093 

6  

7  

8# <<<< PREAMBLE CONTENT >>>> 

9  

10from libc.stdint cimport ( 

11 uint32_t, 

12 uint64_t, 

13 uint8_t, 

14) 

15  

16  

17# <<<< END OF PREAMBLE CONTENT >>>> 

18  

19from ._internal cimport cudla as _cudla 

20  

21  

22  

23  

24############################################################################### 

25# Wrapper functions 

26############################################################################### 

27  

28cdef cudlaStatus cudlaGetVersion(uint64_t* const version) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil: (empty)

29 return _cudla._cudlaGetVersion(version) 

30  

31  

32cdef cudlaStatus cudlaDeviceGetCount(uint64_t* const pNumDevices) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil: (empty)

33 return _cudla._cudlaDeviceGetCount(pNumDevices) 

34  

35  

36cdef cudlaStatus cudlaCreateDevice(const uint64_t device, cudlaDevHandle* const devHandle, const uint32_t flags) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil: (empty)

37 return _cudla._cudlaCreateDevice(device, devHandle, flags) 

38  

39  

40cdef cudlaStatus cudlaMemRegister(const cudlaDevHandle devHandle, const uint64_t* const ptr, const size_t size, uint64_t** const devPtr, const uint32_t flags) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil: (empty)

41 return _cudla._cudlaMemRegister(devHandle, ptr, size, devPtr, flags) 

42  

43  

44cdef cudlaStatus cudlaModuleLoadFromMemory(const cudlaDevHandle devHandle, const uint8_t* const pModule, const size_t moduleSize, cudlaModule* const hModule, const uint32_t flags) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil: (empty)

45 return _cudla._cudlaModuleLoadFromMemory(devHandle, pModule, moduleSize, hModule, flags) 

46  

47  

48cdef cudlaStatus cudlaModuleGetAttributes(const cudlaModule hModule, const cudlaModuleAttributeType attrType, cudlaModuleAttribute* const attribute) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil: (empty)

49 return _cudla._cudlaModuleGetAttributes(hModule, attrType, attribute) 

50  

51  

52cdef cudlaStatus cudlaModuleUnload(const cudlaModule hModule, const uint32_t flags) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil: (empty)

53 return _cudla._cudlaModuleUnload(hModule, flags) 

54  

55  

56cdef cudlaStatus cudlaSubmitTask(const cudlaDevHandle devHandle, const cudlaTask* const ptrToTasks, const uint32_t numTasks, void* const stream, const uint32_t flags) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil: (empty)

57 return _cudla._cudlaSubmitTask(devHandle, ptrToTasks, numTasks, stream, flags) 

58  

59  

60cdef cudlaStatus cudlaDeviceGetAttribute(const cudlaDevHandle devHandle, const cudlaDevAttributeType attrib, cudlaDevAttribute* const pAttribute) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil: (empty)

61 return _cudla._cudlaDeviceGetAttribute(devHandle, attrib, pAttribute) 

62  

63  

64cdef cudlaStatus cudlaMemUnregister(const cudlaDevHandle devHandle, const uint64_t* const devPtr) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil: (empty)

65 return _cudla._cudlaMemUnregister(devHandle, devPtr) 

66  

67  

68cdef cudlaStatus cudlaGetLastError(const cudlaDevHandle devHandle) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil: (empty)

69 return _cudla._cudlaGetLastError(devHandle) 

70  

71  

72cdef cudlaStatus cudlaDestroyDevice(const cudlaDevHandle devHandle) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil: (empty)

73 return _cudla._cudlaDestroyDevice(devHandle) 

74  

75  

76cdef cudlaStatus cudlaSetTaskTimeoutInMs(const cudlaDevHandle devHandle, const uint32_t timeout) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil: (empty)

77 return _cudla._cudlaSetTaskTimeoutInMs(devHandle, timeout)