Refactor dgetrf function to use blocked algorithm

vlang · Apr 28, 2024 · 88776da · 88776da
1 parent e82bda8
commit 88776da
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/lapack/cflags_d_vsl_lapack_lapacke copy.v → lapack/cflags_d_vsl_lapack_lapacke.v b/lapack/cflags_d_vsl_lapack_lapacke copy.v → lapack/cflags_d_vsl_lapack_lapacke.v
diff --git a/lapack/lapack64/dgetrf.v b/lapack/lapack64/dgetrf.v
@@ -73,7 +73,15 @@ pub fn dgetrf(m int, n int, mut a []f64, lda int, ipiv []int) {
 			// apply interchanges to columns 1..j-1.
 			mut slice := unsafe { a[j + jb..] }
 			dlaswp(j, mut slice, lda, j, j + jb, ipiv[..j + jb], 1)
-			//
+
+			blas.dtstrf(.left, .lower, .notrans, .unit, jb, n - j - jb, 1, a[j * lda + j..],
+				lda, a[j * lda + j + jb..], lda)
+
+			if j + jb < m {
+				blas.dgemm(.notrans, .notrans, m - j - jb, n - j - jb, jb, -1, a[(j + jb) * lda + j..],
+					lda, a[j * lda + j + jb..], lda, 1, a[(j + jb) * lda + j + jb..],
+					lda)
+			}
 		}
 	}
 }
diff --git a/ml/knn.v b/ml/knn.v
@@ -112,7 +112,7 @@ pub struct PredictConfig {
 pub:
 	max_iter int
 	k        int
-	to_pred []f64
+	to_pred  []f64
 }
 
 // predict will find the `k` points nearest to the specified `to_pred`.