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: LicenseRef-NVIDIA-SOFTWARE-LICENSE 

3  

4# This code was automatically generated with version 1.5.0, generator version 0.3.1.dev1465+gc5c5c8652. Do not modify it directly. 

5  

6from ._internal cimport cudla as _cudla 

7  

8  

9  

10  

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

12# Wrapper functions 

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

14  

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

16 return _cudla._cudlaGetVersion(version) 

17  

18  

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

20 return _cudla._cudlaDeviceGetCount(pNumDevices) 

21  

22  

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

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

25  

26  

27cdef 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)

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

29  

30  

31cdef 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)

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

33  

34  

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

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

37  

38  

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

40 return _cudla._cudlaModuleUnload(hModule, flags) 

41  

42  

43cdef 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)

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

45  

46  

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

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

49  

50  

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

52 return _cudla._cudlaMemUnregister(devHandle, devPtr) 

53  

54  

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

56 return _cudla._cudlaGetLastError(devHandle) 

57  

58  

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

60 return _cudla._cudlaDestroyDevice(devHandle) 

61  

62  

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

64 return _cudla._cudlaSetTaskTimeoutInMs(devHandle, timeout)