bugfix in mat.cc for cuda
This commit is contained in:
1 parent
4f3ec9aef9
commit
e8979df1b1
1 file changed
+1
-1
@@ -2565,7 +2565,7 @@ const double* NRMat<double>::diagonalof(NRVec<double> &r, const bool divide, boo
|
|||||||
}else{
|
}else{
|
||||||
NRVec<double> tmp(mm, cpu);
|
NRVec<double> tmp(mm, cpu);
|
||||||
for(int i=0;i<mm;i++){
|
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[i] = x;
|
||||||
}
|
}
|
||||||
tmp.moveto(location);
|
tmp.moveto(location);
|
||||||
|
|||||||
Reference in new issue
Block a user