Minor bug fixes

primme · Mar 24, 2024 · 7c55148 · 7c55148
1 parent 2d2e1c1
commit 7c55148
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/eigs/main_iter.c b/src/eigs/main_iter.c
@@ -1666,7 +1666,7 @@ int sketched_main_iter_Sprimme(HEVAL *evals, SCALAR *evecs, PRIMME_INT ldevecs,
    CHKERR(Num_malloc_Sprimme(ldV*primme->maxBasisSize, &V_temp, ctx));
 
    CHKERR(Num_malloc_Rprimme(primme->numEvals, &normalize_evecs, ctx));
-   CHKERR(Num_zero_matrix_Sprimme(T, ldT, ldT, 1, ctx));
+   CHKERR(Num_zero_matrix_Sprimme(T, ldT, ldT, ldT, ctx));
 
 
 

diff --git a/src/eigs/sketch.c b/src/eigs/sketch.c
@@ -162,8 +162,11 @@ int sketch_basis_Sprimme(SCALAR *V, PRIMME_INT ldV, SCALAR *SV, PRIMME_INT ldSV,
 
    /* Find the sketched basis */
    CHKERR(globalSum_Sprimme(&SV[basisSize*ldSV], blockSize*ldSV, ctx));  
-   CHKERR(Num_zero_matrix_Sprimme(&T[basisSize], blockSize, basisSize, ldT, ctx));
-   CHKERR(ortho_Sprimme(SV, ldSV, T, ldT, basisSize, basisSize+blockSize-1, NULL, 0, 0, ldSV, primme->iseed, ctx));
+
+   if(T) {
+      CHKERR(Num_zero_matrix_Sprimme(&T[basisSize], blockSize, basisSize, ldT, ctx));
+      CHKERR(ortho_Sprimme(SV, ldSV, T, ldT, basisSize, basisSize+blockSize-1, NULL, 0, 0, ldSV, primme->iseed, ctx));
+   }
 
    if (primme) primme->stats.timeSketchMatvec += primme_wTimer() - t0;