Add generated MPS enums and structs

JuliaGPU · Dec 10, 2024 · 4b4ed49 · 4b4ed49
1 parent 5ebc3be
commit 4b4ed49
Show file tree

Hide file tree

Showing 9 changed files with 710 additions and 86 deletions.
diff --git a/lib/mps/MPS.jl b/lib/mps/MPS.jl
@@ -20,6 +20,9 @@ const MtlFloat = Union{Float32, Float16}
 
 is_supported(dev::MTLDevice) = ccall(:MPSSupportsMTLDevice, Bool, (id{MTLDevice},), dev)
 
+# Load in generated enums and structs
+include("libmps.jl")
+
 include("size.jl")
 
 # high-level wrappers

diff --git a/lib/mps/copy.jl b/lib/mps/copy.jl
@@ -2,12 +2,12 @@
 
 export MPSMatrixCopyDescriptor
 
-struct MPSMatrixCopyOffsets
-    sourceRowOffset::Cuint
-    sourceColumnOffset::Cuint
-    destinationRowOffset::Cuint
-    destinationColumnOffset::Cuint
-end
+# struct MPSMatrixCopyOffsets
+#     sourceRowOffset::Cuint
+#     sourceColumnOffset::Cuint
+#     destinationRowOffset::Cuint
+#     destinationColumnOffset::Cuint
+# end
 
 @objcwrapper MPSMatrixCopyDescriptor <: NSObject
 

diff --git a/lib/mps/decomposition.jl b/lib/mps/decomposition.jl
@@ -1,9 +1,9 @@
-@cenum MPSMatrixDecompositionStatus::Cint begin
-    MPSMatrixDecompositionStatusSuccess =  0
-    MPSMatrixDecompositionStatusFailure = -1
-    MPSMatrixDecompositionStatusSingular = -2
-    MPSMatrixDecompositionStatusNonPositiveDefinite = -3
-end
+# @cenum MPSMatrixDecompositionStatus::Cint begin
+#     MPSMatrixDecompositionStatusSuccess =  0
+#     MPSMatrixDecompositionStatusFailure = -1
+#     MPSMatrixDecompositionStatusSingular = -2
+#     MPSMatrixDecompositionStatusNonPositiveDefinite = -3
+# end
 
 
 ## lu

diff --git a/lib/mps/images.jl b/lib/mps/images.jl
@@ -1,12 +1,12 @@
 ## kernels
 
-@cenum MPSImageEdgeMode::NSUInteger begin
-    MPSImageEdgeModeZero           = 0
-    MPSImageEdgeModeClamp          = 1
-    MPSImageEdgeModeMirror         = 2
-    MPSImageEdgeModeMirrorWithEdge = 3
-    MPSImageEdgeModeConstant       = 4
-end
+# @cenum MPSImageEdgeMode::NSUInteger begin
+#     MPSImageEdgeModeZero           = 0
+#     MPSImageEdgeModeClamp          = 1
+#     MPSImageEdgeModeMirror         = 2
+#     MPSImageEdgeModeMirrorWithEdge = 3
+#     MPSImageEdgeModeConstant       = 4
+# end
 
 @objcwrapper immutable=false MPSUnaryImageKernel <: MPSKernel
 

diff --git a/lib/mps/kernel.jl b/lib/mps/kernel.jl
@@ -2,14 +2,14 @@
 # kernels
 #
 
-@cenum MPSKernelOptions::NSUInteger begin
-    MPSKernelOptionsNone = 0
-    MPSKernelOptionsSkipAPIValidation = 1 << 0
-    MPSKernelOptionsAllowReducedPrecision = 1 << 1
-    MPSKernelOptionsDisableInternalTiling = 1 << 2
-    MPSKernelOptionsInsertDebugGroups = 1 << 3
-    MPSKernelOptionsVerbose = 1 << 4
-end
+# @cenum MPSKernelOptions::NSUInteger begin
+#     MPSKernelOptionsNone = 0
+#     MPSKernelOptionsSkipAPIValidation = 1 << 0
+#     MPSKernelOptionsAllowReducedPrecision = 1 << 1
+#     MPSKernelOptionsDisableInternalTiling = 1 << 2
+#     MPSKernelOptionsInsertDebugGroups = 1 << 3
+#     MPSKernelOptionsVerbose = 1 << 4
+# end
 
 
 @objcwrapper MPSKernel <: NSObject