add computation of matrix norm in native

Author: devcrocod

multik-core/src/commonMain/kotlin/org/jetbrains/kotlinx/multik/api/linalg/LinAlg.kt

@@ -21,8 +21,8 @@ public interface LinAlg {
      */
     public fun <T : Number> pow(mat: MultiArray<T, D2>, n: Int): NDArray<T, D2>
 
-    /**
-     * Matrix ov vector norm. The default is Frobenius norm.
-     */
-    public fun <T : Number> norm(mat: MultiArray<T, D2>, p: Int = 2): Double
+//    /**
+//     * Matrix ov vector norm. The default is Frobenius norm.
+//     */
+//    public fun <T : Number> norm(mat: MultiArray<T, D2>, p: Int = 2): Double
 }

multik-core/src/commonMain/kotlin/org/jetbrains/kotlinx/multik/api/linalg/LinAlgEx.kt

@@ -21,6 +21,10 @@ public interface LinAlgEx {
     public fun <D : Dim2> solveF(a: MultiArray<Float, D2>, b: MultiArray<Float, D>): NDArray<Float, D>
     public fun <T : Complex, D : Dim2> solveC(a: MultiArray<T, D2>, b: MultiArray<T, D>): NDArray<T, D>
 
+    public fun normF(mat: MultiArray<Float, D2>, norm: Norm = Norm.Fro): Float
+
+    public fun norm(mat: MultiArray<Double, D2>, norm: Norm = Norm.Fro): Double
+
     public fun <T : Number> qr(mat: MultiArray<T, D2>): Pair<D2Array<Double>, D2Array<Double>>
     public fun qrF(mat: MultiArray<Float, D2>): Pair<D2Array<Float>, D2Array<Float>>
     public fun <T : Complex> qrC(mat: MultiArray<T, D2>): Pair<D2Array<T>, D2Array<T>>

multik-core/src/commonMain/kotlin/org/jetbrains/kotlinx/multik/api/linalg/NormClassification.kt

@@ -0,0 +1,8 @@
+package org.jetbrains.kotlinx.multik.api.linalg
+
+public enum class Norm(public val lapackCode: Char) {
+    Max('M'),
+    N1('1'),
+    Inf('I'),
+    Fro('F')
+}

multik-core/src/commonMain/kotlin/org/jetbrains/kotlinx/multik/api/linalg/norm.kt

@@ -0,0 +1,17 @@
+package org.jetbrains.kotlinx.multik.api.linalg
+
+import org.jetbrains.kotlinx.multik.ndarray.data.D2
+import org.jetbrains.kotlinx.multik.ndarray.data.MultiArray
+import kotlin.jvm.JvmName
+
+/**
+ * Returns norm of float matrix
+ */
+@JvmName("normF")
+public fun LinAlg.norm(mat: MultiArray<Float, D2>, norm: Norm = Norm.Fro): Float = this.linAlgEx.normF(mat, norm)
+
+/**
+ * Returns norm of float matrix
+ */
+@JvmName("normD")
+public fun LinAlg.norm(mat: MultiArray<Double, D2>, norm: Norm = Norm.Fro): Double = this.linAlgEx.norm(mat, norm)

multik-default/src/commonMain/kotlin/org/jetbrains/kotlinx/multik/default/linalg/DefaultLinAlg.kt

@@ -15,6 +15,4 @@ public expect object DefaultLinAlg : LinAlg {
     override val linAlgEx: LinAlgEx
 
     override fun <T : Number> pow(mat: MultiArray<T, D2>, n: Int): NDArray<T, D2>
-
-    override fun <T : Number> norm(mat: MultiArray<T, D2>, p: Int): Double
 }

multik-default/src/commonMain/kotlin/org/jetbrains/kotlinx/multik/default/linalg/DefaultLinAlgEx.kt

@@ -5,6 +5,7 @@
 package org.jetbrains.kotlinx.multik.default.linalg
 
 import org.jetbrains.kotlinx.multik.api.linalg.LinAlgEx
+import org.jetbrains.kotlinx.multik.api.linalg.Norm
 import org.jetbrains.kotlinx.multik.ndarray.complex.Complex
 import org.jetbrains.kotlinx.multik.ndarray.complex.ComplexDouble
 import org.jetbrains.kotlinx.multik.ndarray.complex.ComplexFloat
@@ -21,6 +22,8 @@ public expect object DefaultLinAlgEx: LinAlgEx {
 
     override fun <D : Dim2> solveF(a: MultiArray<Float, D2>, b: MultiArray<Float, D>): NDArray<Float, D>
     override fun <T : Complex, D : Dim2> solveC(a: MultiArray<T, D2>, b: MultiArray<T, D>): NDArray<T, D>
+    override fun normF(mat: MultiArray<Float, D2>, norm: Norm): Float
+    override fun norm(mat: MultiArray<Double, D2>, norm: Norm): Double
     override fun <T : Number> qr(mat: MultiArray<T, D2>): Pair<D2Array<Double>, D2Array<Double>>
     override fun qrF(mat: MultiArray<Float, D2>): Pair<D2Array<Float>, D2Array<Float>>
     override fun <T : Complex> qrC(mat: MultiArray<T, D2>): Pair<D2Array<T>, D2Array<T>>

multik-default/src/iosMain/kotlin/org.jetbrains.kotlinx.multik.default/linalg/DefaultLinAlg.kt

@@ -17,6 +17,4 @@ public actual object DefaultLinAlg : LinAlg {
         get() = DefaultLinAlgEx
 
     actual override fun <T : Number> pow(mat: MultiArray<T, D2>, n: Int): NDArray<T, D2> = KELinAlg.pow(mat, n)
-
-    actual override fun <T : Number> norm(mat: MultiArray<T, D2>, p: Int): Double = KELinAlg.norm(mat, p)
 }

multik-default/src/iosMain/kotlin/org.jetbrains.kotlinx.multik.default/linalg/DefaultLinAlgEx.kt

@@ -5,6 +5,7 @@
 package org.jetbrains.kotlinx.multik.default.linalg
 
 import org.jetbrains.kotlinx.multik.api.linalg.LinAlgEx
+import org.jetbrains.kotlinx.multik.api.linalg.Norm
 import org.jetbrains.kotlinx.multik.api.linalg.dot
 import org.jetbrains.kotlinx.multik.kotlin.linalg.KELinAlg
 import org.jetbrains.kotlinx.multik.kotlin.linalg.KELinAlgEx
@@ -30,6 +31,10 @@ public actual object DefaultLinAlgEx : LinAlgEx {
     actual override fun <T : Complex, D : Dim2> solveC(a: MultiArray<T, D2>, b: MultiArray<T, D>): NDArray<T, D> =
         KELinAlgEx.solveC(a, b)
 
+    actual override fun normF(mat: MultiArray<Float, D2>, norm: Norm): Float = KELinAlgEx.normF(mat, norm)
+
+    actual override fun norm(mat: MultiArray<Double, D2>, norm: Norm): Double = KELinAlgEx.norm(mat, norm)
+
     actual override fun <T : Number> qr(mat: MultiArray<T, D2>): Pair<D2Array<Double>, D2Array<Double>> =
         KELinAlgEx.qr(mat)
 

multik-default/src/jsMain/kotlin/org.jetbrains.kotlinx.multik.default/linalg/DefaultLinAlg.kt

@@ -17,6 +17,4 @@ public actual object DefaultLinAlg : LinAlg {
         get() = DefaultLinAlgEx
 
     actual override fun <T : Number> pow(mat: MultiArray<T, D2>, n: Int): NDArray<T, D2> = KELinAlg.pow(mat, n)
-
-    actual override fun <T : Number> norm(mat: MultiArray<T, D2>, p: Int): Double = KELinAlg.norm(mat, p)
 }

multik-default/src/jsMain/kotlin/org.jetbrains.kotlinx.multik.default/linalg/DefaultLinAlgEx.kt

@@ -5,6 +5,7 @@
 package org.jetbrains.kotlinx.multik.default.linalg
 
 import org.jetbrains.kotlinx.multik.api.linalg.LinAlgEx
+import org.jetbrains.kotlinx.multik.api.linalg.Norm
 import org.jetbrains.kotlinx.multik.api.linalg.dot
 import org.jetbrains.kotlinx.multik.kotlin.linalg.KELinAlg
 import org.jetbrains.kotlinx.multik.kotlin.linalg.KELinAlgEx
@@ -29,6 +30,10 @@ public actual object DefaultLinAlgEx : LinAlgEx {
     actual override fun <T : Complex, D : Dim2> solveC(a: MultiArray<T, D2>, b: MultiArray<T, D>): NDArray<T, D> =
         KELinAlgEx.solveC(a, b)
 
+    actual override fun normF(mat: MultiArray<Float, D2>, norm: Norm): Float = KELinAlgEx.normF(mat, norm)
+
+    actual override fun norm(mat: MultiArray<Double, D2>, norm: Norm): Double = KELinAlgEx.norm(mat, norm)
+
     actual override fun <T : Number> qr(mat: MultiArray<T, D2>): Pair<D2Array<Double>, D2Array<Double>> =
         KELinAlgEx.qr(mat)