Mentions légales du service

Skip to content
Snippets Groups Projects
Commit 71848245 authored by Mathieu Faverge's avatar Mathieu Faverge
Browse files

compute/map: Fix the pmap function for pentagonal matrices

parent 0cc69473
No related branches found
No related tags found
1 merge request!445Mapv: Add some missing fixes, and make the map kernel non runtime dependent.
...@@ -11,7 +11,7 @@ ...@@ -11,7 +11,7 @@
* *
* @version 1.3.0 * @version 1.3.0
* @author Mathieu Faverge * @author Mathieu Faverge
* @date 2024-03-11 * @date 2024-03-14
* *
*/ */
#include "control/common.h" #include "control/common.h"
...@@ -28,31 +28,34 @@ void chameleon_pmap( cham_uplo_t uplo, int ndata, cham_map_data_t *data, ...@@ -28,31 +28,34 @@ void chameleon_pmap( cham_uplo_t uplo, int ndata, cham_map_data_t *data,
CHAM_context_t *chamctxt; CHAM_context_t *chamctxt;
RUNTIME_option_t options; RUNTIME_option_t options;
const CHAM_desc_t *A = data[0].desc; const CHAM_desc_t *A = data[0].desc;
int m, n; int m, n, minmn;
chamctxt = chameleon_context_self(); chamctxt = chameleon_context_self();
if (sequence->status != CHAMELEON_SUCCESS) if (sequence->status != CHAMELEON_SUCCESS)
return; return;
RUNTIME_options_init( &options, chamctxt, sequence, request ); RUNTIME_options_init( &options, chamctxt, sequence, request );
minmn = chameleon_min( A->mt, A->nt );
switch( uplo ) { switch( uplo ) {
case ChamUpper: case ChamUpper:
for (n = 0; n < A->nt; n++) { for (m = 0; m < minmn; m++) {
for (m = 0; m < n; m++) { INSERT_TASK_map(
&options, uplo, m, m,
ndata, data,
op_fct, op_args );
for (n = m+1; n < A->nt; n++) {
INSERT_TASK_map( INSERT_TASK_map(
&options, ChamUpperLower, m, n, &options, ChamUpperLower, m, n,
ndata, data, ndata, data,
op_fct, op_args ); op_fct, op_args );
} }
INSERT_TASK_map(
&options, uplo, n, n,
ndata, data,
op_fct, op_args );
} }
break; break;
case ChamLower: case ChamLower:
for (n = 0; n < A->nt; n++) { for (n = 0; n < minmn; n++){
INSERT_TASK_map( INSERT_TASK_map(
&options, uplo, n, n, &options, uplo, n, n,
ndata, data, ndata, data,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment