From 3e18e5923bd6a7fe012b869e24a3a7fa44a8322b Mon Sep 17 00:00:00 2001
From: Mathieu Faverge <mathieu.faverge@inria.fr>
Date: Fri, 13 Dec 2024 18:33:21 +0100
Subject: [PATCH] compute/gepdf: cuda QR call MUST have the diagonal copy to be
valid
---
compute/zgepdf_qr.c | 32 +++++++++++++++++++++++++++-----
1 file changed, 27 insertions(+), 5 deletions(-)
diff --git a/compute/zgepdf_qr.c b/compute/zgepdf_qr.c
index 35622025f..d138b0517 100644
--- a/compute/zgepdf_qr.c
+++ b/compute/zgepdf_qr.c
@@ -82,10 +82,12 @@ int CHAMELEON_zgepdf_qr_Tile( int doqr, int optid,
CHAM_desc_t *A1, CHAM_desc_t *TS1, CHAM_desc_t *TT1, CHAM_desc_t *Q1,
CHAM_desc_t *A2, CHAM_desc_t *TS2, CHAM_desc_t *TT2, CHAM_desc_t *Q2 )
{
- CHAM_context_t *chamctxt;
+ CHAM_context_t *chamctxt;
RUNTIME_sequence_t *sequence = NULL;
- RUNTIME_request_t request = RUNTIME_REQUEST_INITIALIZER;
- int status;
+ RUNTIME_request_t request = RUNTIME_REQUEST_INITIALIZER;
+ CHAM_desc_t D1, *D1ptr = NULL;
+ CHAM_desc_t D2, *D2ptr = NULL;
+ int status;
chamctxt = chameleon_context_self();
if (chamctxt == NULL) {
@@ -94,16 +96,36 @@ int CHAMELEON_zgepdf_qr_Tile( int doqr, int optid,
}
chameleon_sequence_create( chamctxt, &sequence );
+#if defined(CHAMELEON_COPY_DIAG)
+ {
+ int n = A1->n;
+ chameleon_zdesc_copy_and_restrict( A1, &D1, A1->m, n );
+ D1ptr = &D1;
+ chameleon_zdesc_copy_and_restrict( A2, &D2, A2->m, n );
+ D2ptr = &D2;
+ }
+#endif
+
chameleon_pzgepdf_qr( 1, doqr, optid, qrtreeT, qrtreeB,
- A1, TS1, TT1, NULL, Q1,
- A2, TS2, TT2, NULL, Q2,
+ A1, TS1, TT1, D1ptr, Q1,
+ A2, TS2, TT2, D2ptr, Q2,
sequence, &request );
CHAMELEON_Desc_Flush( Q1, sequence );
CHAMELEON_Desc_Flush( Q2, sequence );
+ if ( D1ptr != NULL ) {
+ CHAMELEON_Desc_Flush( D1ptr, sequence );
+ CHAMELEON_Desc_Flush( D2ptr, sequence );
+ }
chameleon_sequence_wait( chamctxt, sequence );
status = sequence->status;
chameleon_sequence_destroy( chamctxt, sequence );
+
+ if ( D1ptr != NULL ) {
+ chameleon_desc_destroy( D1ptr );
+ chameleon_desc_destroy( D2ptr );
+ }
+
return status;
}
--
GitLab