From e8979df1b1d30711dcb3cb785235fa30a60c854e Mon Sep 17 00:00:00 2001
From: Jiri Pittner <jiri@pittnerovi.com>
Date: Thu, 29 Oct 2026 23:46:00 +0100
Subject: [PATCH] bugfix in mat.cc for cuda

---
 mat.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mat.cc b/mat.cc
index 84d608f..cff0d66 100644
--- a/mat.cc
+++ b/mat.cc
@@ -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);