bugfix in mat.cc for cuda

This commit is contained in:
jiri committed 2026-10-29 23:46:00 +01:00
1 parent 4f3ec9aef9
commit e8979df1b1
1 file changed
+1 -1
+1 -1
View File
@@ -2565,7 +2565,7 @@ const double* NRMat<double>::diagonalof(NRVec<double> &r, const bool divide, boo
}else{
NRVec<double> tmp(mm, cpu);
for(int i=0;i<mm;i++){
const double x = cublasDdot(nn, v + i, 1, v + i, 1);
const double x = cublasDdot(nn, v + i, mm, v + i, mm); //column i
tmp[i] = x;
}
tmp.moveto(location);