Start preparing for versioned DLPack

pabloferz · May 28, 2024 · e1b7c6f · e1b7c6f
1 parent b8e5c90
commit e1b7c6f
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/DLPack.jl b/src/DLPack.jl
@@ -22,6 +22,11 @@ using Requires
 
 ##  Types  ##
 
+struct DLPackVersion
+    major::Cuint
+    minor::Cuint
+end
+
 @enum DLDeviceType::Cint begin
     kDLCPU = 1
     kDLCUDA = 2
@@ -110,6 +115,14 @@ mutable struct DLManagedTensor
     end
 end
 
+mutable struct DLManagedTensorVersioned
+    version::DLPackVersion
+    manager_ctx::Ptr{Cvoid}
+    deleter::Ptr{Cvoid}
+    flags::Culonglong
+    dl_tensor::DLTensor
+end
+
 """
     Capsule