From f477bf60913b9760235ac6e88540f3874d7f7442 Mon Sep 17 00:00:00 2001
From: Mathieu Faverge <mathieu.faverge@inria.fr>
Date: Wed, 5 Apr 2017 17:15:29 +0200
Subject: [PATCH] Be more consistant with the documentation and prevent problem
if kdwork changes for stricter arrays
---
cudablas/compute/cuda_ztpmqrt.c | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/cudablas/compute/cuda_ztpmqrt.c b/cudablas/compute/cuda_ztpmqrt.c
index 640f9fdc8..350dc3df8 100644
--- a/cudablas/compute/cuda_ztpmqrt.c
+++ b/cudablas/compute/cuda_ztpmqrt.c
@@ -48,14 +48,14 @@ CUDA_ztpmqrt( MORSE_enum side, MORSE_enum trans,
n1 = N;
ldwork = IB;
ldworkc = M;
- ws = IB * n1;
+ ws = ldwork * n1;
}
else {
m1 = M;
n1 = K;
ldwork = m1;
ldworkc = IB;
- ws = IB * m1;
+ ws = ldwork * IB;
}
/* TS case */
--
GitLab