Coverage for cuda/core/_dlpack.pyx: 92.56%

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

2# 

3# SPDX-License-Identifier: Apache-2.0 

4  

5  

6from enum import IntEnum (empty)

7  

8  

9cdef void pycapsule_deleter(object capsule) noexcept: (empty)

10 cdef DLManagedTensor* dlm_tensor 

11 cdef DLManagedTensorVersioned* dlm_tensor_ver 

12 # Do not invoke the deleter on a used capsule. 

13 if cpython.PyCapsule_IsValid( 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

14 capsule, DLPACK_TENSOR_UNUSED_NAME): 

15 dlm_tensor = <DLManagedTensor*>( 1 ctx1a

16 cpython.PyCapsule_GetPointer( 1 ctx1a

17 capsule, DLPACK_TENSOR_UNUSED_NAME)) 

18 if dlm_tensor.deleter: 1 ctx1a

19 dlm_tensor.deleter(dlm_tensor) 1 ctx1a

20 elif cpython.PyCapsule_IsValid( 112 ctx2xbb c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

21 capsule, DLPACK_VERSIONED_TENSOR_UNUSED_NAME): 

22 dlm_tensor_ver = <DLManagedTensorVersioned*>( 1 ctx1a

23 cpython.PyCapsule_GetPointer( 1 ctx1a

24 capsule, DLPACK_VERSIONED_TENSOR_UNUSED_NAME)) 

25 if dlm_tensor_ver.deleter: 1 ctx1a

26 dlm_tensor_ver.deleter(dlm_tensor_ver) 2 ctx2xba

27  

28  

29cdef void deleter(DLManagedTensor* tensor) noexcept with gil: (empty)

30 if tensor: 2 ctx2wba

31 if tensor.dl_tensor.shape: 2 ctx2wba

32 stdlib.free(tensor.dl_tensor.shape) 2 ctx2wba

33 if tensor.manager_ctx: 2 ctx2wba

34 cpython.Py_DECREF(<object>tensor.manager_ctx) 2 ctx2wba

35 tensor.manager_ctx = NULL 2 ctx2wba

36 stdlib.free(tensor) 2 ctx2wba

37  

38  

39cdef void versioned_deleter(DLManagedTensorVersioned* tensor) noexcept with gil: (empty)

40 if tensor: 112 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwba qbrbsbtbubvb

41 if tensor.dl_tensor.shape: 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

42 stdlib.free(tensor.dl_tensor.shape) 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

43 if tensor.manager_ctx: 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

44 cpython.Py_DECREF(<object>tensor.manager_ctx) 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

45 tensor.manager_ctx = NULL 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

46 stdlib.free(tensor) 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

47  

48  

49cdef inline DLManagedTensorVersioned* allocate_dlm_tensor_versioned() except? NULL: (empty)

50 cdef DLManagedTensorVersioned* dlm_tensor_ver = NULL 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

51 try: 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

52 dlm_tensor_ver = <DLManagedTensorVersioned*>( 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

53 stdlib.malloc(sizeof(DLManagedTensorVersioned))) 

54 dlm_tensor_ver.dl_tensor.shape = NULL 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

55 dlm_tensor_ver.manager_ctx = NULL 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

56 return dlm_tensor_ver 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

57 except: 

58 if dlm_tensor_ver: 

59 stdlib.free(dlm_tensor_ver) 

60 raise 

61  

62  

63cdef inline DLManagedTensor* allocate_dlm_tensor() except? NULL: (empty)

64 cdef DLManagedTensor* dlm_tensor = NULL 2 ctx2wba

65 try: 2 ctx2wba

66 dlm_tensor = <DLManagedTensor*>( 2 ctx2wba

67 stdlib.malloc(sizeof(DLManagedTensor))) 

68 dlm_tensor.dl_tensor.shape = NULL 2 ctx2wba

69 dlm_tensor.manager_ctx = NULL 2 ctx2wba

70 return dlm_tensor 2 ctx2wba

71 except: 

72 if dlm_tensor: 

73 stdlib.free(dlm_tensor) 

74 raise 

75  

76  

77cdef inline int setup_dl_tensor_layout(DLTensor* dl_tensor, object buf) except -1: (empty)

78 dl_tensor.ndim = 1 112 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwba qbrbsbtbubvb

79 cdef int64_t* shape_strides = \ 

80 <int64_t*>stdlib.malloc(sizeof(int64_t) * 2) 112 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwba qbrbsbtbubvb

81 if shape_strides == NULL: 112 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwba qbrbsbtbubvb

82 raise MemoryError() 

83 # DLPack v1.2+ requires non-NULL strides for ndim != 0. 

84 shape_strides[0] = <int64_t>buf.size 112 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwba qbrbsbtbubvb

85 shape_strides[1] = 1 112 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwba qbrbsbtbubvb

86 dl_tensor.shape = shape_strides 112 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwba qbrbsbtbubvb

87 dl_tensor.strides = shape_strides + 1 112 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwba qbrbsbtbubvb

88 dl_tensor.byte_offset = 0 112 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwba qbrbsbtbubvb

89 return 0 112 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwba qbrbsbtbubvb

90  

91  

92def classify_dl_device(buf: object) -> tuple[int, int]: (empty)

93 """Classify a buffer into a DLPack (device_type, device_id) pair. 

94  

95 ``buf`` must expose ``is_device_accessible``, ``is_host_accessible``, 

96 ``is_managed``, and ``device_id`` attributes. 

97 """ 

98 cdef bint d = buf.is_device_accessible 118 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwbybCbzbAbBba qbrbsbtbubDbvb

99 cdef bint h = buf.is_host_accessible 118 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwbybCbzbAbBba qbrbsbtbubDbvb

100 if d and not h: 118 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwbybCbzbAbBba qbrbsbtbubDbvb

101 return (_kDLCUDA, buf.device_id) 4 ctx2Cba Dbvb

102 if d and h: 114 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwbybzbAbBbqbrbsbtbub

103 return (_kDLCUDAManaged if buf.is_managed else _kDLCUDAHost, 0) 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbAbBbqbrbsbtbub

104 if not d and h: 3 ctx2wbybzb

105 return (_kDLCPU, 0) 1 ctx2zb

106 raise BufferError("buffer is neither device-accessible nor host-accessible") 2 ctx2wbyb

107  

108  

109cdef inline int setup_dl_tensor_device(DLTensor* dl_tensor, object buf) except -1: (empty)

110 cdef DLDevice* device = &dl_tensor.device 112 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwba qbrbsbtbubvb

111 dev_type, dev_id = classify_dl_device(buf) 112 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwba qbrbsbtbubvb

112 device.device_type = <_DLDeviceType>dev_type 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

113 device.device_id = <int32_t>dev_id 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

114 return 0 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

115  

116  

117cdef inline int setup_dl_tensor_dtype(DLTensor* dl_tensor) except -1 nogil: (empty)

118 dl_tensor.dtype = DLDataType( 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

119 code=<uint8_t>kDLInt, bits=<uint8_t>8, lanes=<uint16_t>1) 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

120 return 0 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

121  

122  

123cpdef object make_py_capsule(object buf, bint versioned): (empty)

124 cdef DLManagedTensor* dlm_tensor = NULL 112 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwba qbrbsbtbubvb

125 cdef DLManagedTensorVersioned* dlm_tensor_ver = NULL 112 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwba qbrbsbtbubvb

126 cdef DLTensor* dl_tensor 

127 cdef void* tensor_ptr 

128 cdef const char* capsule_name 

129 cdef object ret = None 112 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwba qbrbsbtbubvb

130  

131 try: 112 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwba qbrbsbtbubvb

132 if versioned: 112 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwba qbrbsbtbubvb

133 dlm_tensor_ver = allocate_dlm_tensor_versioned() 112 ctx2xbb c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

134 # Transfer the reference to manager_ctx 

135 cpython.Py_INCREF(buf) 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

136 dlm_tensor_ver.manager_ctx = <void*>buf 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

137 dlm_tensor_ver.deleter = versioned_deleter 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

138 dlm_tensor_ver.version.major = DLPACK_MAJOR_VERSION 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

139 dlm_tensor_ver.version.minor = DLPACK_MINOR_VERSION 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

140 dlm_tensor_ver.flags = 0 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

141 dl_tensor = &dlm_tensor_ver.dl_tensor 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

142 tensor_ptr = dlm_tensor_ver 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

143 capsule_name = DLPACK_VERSIONED_TENSOR_UNUSED_NAME 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

144 else: 

145 dlm_tensor = allocate_dlm_tensor() 2 ctx2wba

146 # Transfer the reference to manager_ctx 

147 cpython.Py_INCREF(buf) 2 ctx2wba

148 dlm_tensor.manager_ctx = <void*>buf 2 ctx2wba

149 dlm_tensor.deleter = deleter 2 ctx2wba

150 dl_tensor = &dlm_tensor.dl_tensor 2 ctx2wba

151 tensor_ptr = dlm_tensor 2 ctx2wba

152 capsule_name = DLPACK_TENSOR_UNUSED_NAME 2 ctx2wba

153  

154 dl_tensor.data = <void*><intptr_t>(int(buf.handle)) 112 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwba qbrbsbtbubvb

155 setup_dl_tensor_layout(dl_tensor, buf) 112 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwba qbrbsbtbubvb

156 setup_dl_tensor_device(dl_tensor, buf) 112 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbwba qbrbsbtbubvb

157 setup_dl_tensor_dtype(dl_tensor) 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

158 ret = cpython.PyCapsule_New(tensor_ptr, capsule_name, pycapsule_deleter) 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

159 except: 1 ctx2wb

160 if ret is None: 1 ctx2wb

161 deleter(dlm_tensor) 1 ctx2wb

162 versioned_deleter(dlm_tensor_ver) 1 ctx2wb

163 raise 1 ctx2wb

164 return ret 111 ctx2b c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpba qbrbsbtbubvb

165  

166  

167# Values are fixed by the DLPack spec; see _include/dlpack.h. They are 

168# hard-coded here (rather than referencing the cdef extern names) so that the 

169# generated .pyi stub doesn't reference Cython-only identifiers. 

170class DLDeviceType(IntEnum): (empty)

171 kDLCPU = 1 (empty)

172 kDLCUDA = 2 (empty)

173 kDLCUDAHost = 3 (empty)

174 kDLCUDAManaged = 13 (empty)