add solvers

Author: tgymnich

lib/mps/MPS.jl

@@ -39,10 +39,10 @@ include("vector.jl")
 include("matrixrandom.jl")
 include("ndarray.jl")
 include("decomposition.jl")
+include("solve.jl")
 include("copy.jl")
 
 # integrations
 include("random.jl")
 include("linalg.jl")
-
 end

lib/mps/libmps.jl

@@ -535,11 +535,11 @@ end
 
 @objcwrapper immutable = false MPSMatrixVectorMultiplication <: MPSMatrixBinaryKernel
 
-@objcwrapper immutable = true MPSMatrixSolveTriangular <: MPSMatrixBinaryKernel
+@objcwrapper immutable = false MPSMatrixSolveTriangular <: MPSMatrixBinaryKernel
 
-@objcwrapper immutable = true MPSMatrixSolveLU <: MPSMatrixBinaryKernel
+@objcwrapper immutable = false MPSMatrixSolveLU <: MPSMatrixBinaryKernel
 
-@objcwrapper immutable = true MPSMatrixSolveCholesky <: MPSMatrixBinaryKernel
+@objcwrapper immutable = false MPSMatrixSolveCholesky <: MPSMatrixBinaryKernel
 
 @cenum MPSMatrixDecompositionStatus::Int32 begin
     MPSMatrixDecompositionStatusSuccess = 0

lib/mps/solve.jl

@@ -0,0 +1,77 @@
+
+export MPSMatrixSolveTriangular
+
+# @objcwrapper immutable=false MPSMatrixSolveTriangular <: MPSMatrixUnaryKernel
+
+function MPSMatrixSolveTriangular(device, right, upper, unit, order, numberOfRightHandSides, alpha)
+    kernel = @objc [MPSMatrixSolveTriangular alloc]::id{MPSMatrixSolveTriangular}
+    obj = MPSMatrixSolveTriangular(kernel)
+    finalizer(release, obj)
+    @objc [obj::id{MPSMatrixSolveTriangular} initWithDevice:device::id{MTLDevice}
+                                             right:right::Bool
+                                             upper:upper::Bool
+                                             transpose:transpose::Bool
+                                             unit:unit::Bool
+                                             order:order::NSUInteger
+                                             numberOfRightHandSides:numberOfRightHandSides::NSUInteger
+                                             alpha:alpha::Float64]::id{MPSMatrixSolveTriangular}
+    return obj
+end
+
+function encode!(cmdbuf::MTLCommandBuffer, kernel::MPSMatrixSolveTriangular, sourceMatrix, resultMatrix, pivotIndices, status)
+    @objc [kernel::id{MPSMatrixSolveTriangular} encodeToCommandBuffer:cmdbuf::id{MTLCommandBuffer}
+                                                sourceMatrix:sourceMatrix::id{MPSMatrix}
+                                                resultMatrix:resultMatrix::id{MPSMatrix}
+                                                pivotIndices:pivotIndices::id{MPSMatrix}
+                                                status:status::id{MPSMatrix}]::Nothing
+end
+
+
+export MPSMatrixSolveLU
+
+# @objcwrapper immutable=false MPSMatrixSolveLU <: MPSMatrixUnaryKernel
+
+function MPSMatrixSolveLU(device, transpose, order, numberOfRightHandSides)
+    kernel = @objc [MPSMatrixSolveLU alloc]::id{MPSMatrixSolveLU}
+    obj = MPSMatrixSolveLU(kernel)
+    finalizer(release, obj)
+    @objc [obj::id{MPSMatrixSolveLU} initWithDevice:device::id{MTLDevice}
+                                             transpose:transpose::Bool
+                                             order:order::NSUInteger
+                                             numberOfRightHandSides:numberOfRightHandSides::NSUInteger]::id{MPSMatrixSolveLU}
+    return obj
+end
+
+function encode!(cmdbuf::MTLCommandBuffer, kernel::MPSMatrixSolveLU, sourceMatrix, rightHandSideMatrix, pivotIndices, solutionMatrix)
+    @objc [kernel::id{MPSMatrixSolveLU} encodeToCommandBuffer:cmdbuf::id{MTLCommandBuffer}
+                                                sourceMatrix:sourceMatrix::id{MPSMatrix}
+                                                rightHandSideMatrix:rightHandSideMatrix::id{MPSMatrix}
+                                                pivotIndices:pivotIndices::id{MPSMatrix}
+                                                solutionMatrix:solutionMatrix::id{MPSMatrix}]::Nothing
+end
+
+
+
+
+export MPSMatrixSolveCholesky
+
+# @objcwrapper immutable=false MPSMatrixSolveCholesky <: MPSMatrixUnaryKernel
+
+function MPSMatrixSolveCholesky(device, upper, order, numberOfRightHandSides)
+    kernel = @objc [MPSMatrixSolveCholesky alloc]::id{MPSMatrixSolveCholesky}
+    obj = MPSMatrixSolveCholesky(kernel)
+    finalizer(release, obj)
+    @objc [obj::id{MPSMatrixSolveCholesky} initWithDevice:device::id{MTLDevice}
+                                             upper:upper::Bool
+                                             order:order::NSUInteger
+                                             numberOfRightHandSides:numberOfRightHandSides::NSUInteger]::id{MPSMatrixSolveCholesky}
+    return obj
+end
+
+function encode!(cmdbuf::MTLCommandBuffer, kernel::MPSMatrixSolveCholesky, sourceMatrix, rightHandSideMatrix, solutionMatrix)
+    @objc [kernel::id{MPSMatrixSolveCholesky} encodeToCommandBuffer:cmdbuf::id{MTLCommandBuffer}
+                                                sourceMatrix:sourceMatrix::id{MPSMatrix}
+                                                rightHandSideMatrix:rightHandSideMatrix::id{MPSMatrix}
+                                                pivotIndices:pivotIndices::id{MPSMatrix}
+                                                solutionMatrix:solutionMatrix::id{MPSMatrix}]::Nothing
+end

res/wrap/libmps.toml

@@ -102,6 +102,15 @@ immutable=false
 [api.MPSMatrixSoftMax]
 immutable=false
 
+[api.MPSMatrixSolveTriangular]
+immutable=false
+
+[api.MPSMatrixSolveLU]
+immutable=false
+
+[api.MPSMatrixSolveCholesky]
+immutable=false
+
 [api.MPSMatrixUnaryKernel]
 immutable=false