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  

6# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=b134ed8cb83eb5a5c6ff30be817e64d09d421d7257761e58fbc06b131690a392 

7from ._internal cimport cudla as _cudla 

8  

9  

10  

11  

12############################################################################### 

13# Wrapper functions 

14############################################################################### 

15  

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

17 return _cudla._cudlaGetVersion(version) 

18  

19  

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

21 return _cudla._cudlaDeviceGetCount(pNumDevices) 

22  

23  

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

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

26  

27  

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

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

30  

31  

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

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

34  

35  

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

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

38  

39  

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

41 return _cudla._cudlaModuleUnload(hModule, flags) 

42  

43  

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

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

46  

47  

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

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

50  

51  

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

53 return _cudla._cudlaMemUnregister(devHandle, devPtr) 

54  

55  

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

57 return _cudla._cudlaGetLastError(devHandle) 

58  

59  

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

61 return _cudla._cudlaDestroyDevice(devHandle) 

62  

63  

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

65 return _cudla._cudlaSetTaskTimeoutInMs(devHandle, timeout)