fixed PCA disagreement

Colin Heinzmann
1 parent 2f6387f4
Showing 1 changed file with 4 additions and 3 deletions
openbr/plugins/cuda/cudapca.cu
@@ -60,12 +60,13 @@ namespace br { namespace cuda { namespace pca {
       return;
     }
+    if (numSteps*stepSize+evIdx >= numEigenvectors) {
+      numSteps--;
+    }
+
     float acc = 0;
     for (int i=0; i < numSteps; i++) {
       int ibIdx = i*stepSize + evIdx;
-      if (ibIdx >= numSteps*stepSize) {
-        break;
-      }
       acc += intermediaryBuffer[ibIdx];
     }