6 #ifndef DLPACK_DLPACK_H_ 7 #define DLPACK_DLPACK_H_ 10 #define DLPACK_EXTERN_C extern "C" 12 #define DLPACK_EXTERN_C 16 #define DLPACK_VERSION 020 21 #define DLPACK_DLL __declspec(dllexport) 23 #define DLPACK_DLL __declspec(dllimport) 172 #endif // DLPACK_DLPACK_H_
Pinned CUDA GPU device by cudaMallocHost.
A Device context for Tensor and operator.
int64_t * strides
strides of the tensor, can be NULL, indicating tensor is compact.
void(* deleter)(struct DLManagedTensor *self)
Destructor signature void (*)(void*) - this should be called to destruct manager_ctx which holds the ...
struct DLManagedTensor DLManagedTensor
C Tensor object, manage memory of DLTensor. This data structure is intended to faciliate the borrowin...
uint8_t code
Type code of base types. We keep it uint8_t instead of DLDataTypeCode for minimal memory footprint...
int device_id
The device index.
int64_t * shape
The shape of the tensor.
C Tensor object, manage memory of DLTensor. This data structure is intended to faciliate the borrowin...
Vulkan buffer for next generation graphics.
uint8_t bits
Number of bits, common choices are 8, 16, 32.
void * data
The opaque data pointer points to the allocated data. This will be CUDA device pointer or cl_mem hand...
DLTensor dl_tensor
DLTensor which is being memory managed.
DLDataType dtype
The data type of the pointer.
DLDeviceType
The device type in DLContext.
DLDeviceType device_type
The device type used in the device.
DLDataTypeCode
The type code options DLDataType.
int ndim
Number of dimensions.
void * manager_ctx
the context of the original host framework of DLManagedTensor in which DLManagedTensor is used in the...
DLContext ctx
The device context of the tensor.
Reserved extension device type, used for quickly test extension device The semantics can differ depen...
The data type the tensor can hold.
uint16_t lanes
Number of lanes in the type, used for vector types.
Plain C Tensor object, does not manage memory.
uint64_t byte_offset
The offset in bytes to the beginning pointer to data.
Verilog simulator buffer.