Mentions légales du service

Skip to content
Snippets Groups Projects

SUMMA GEMM

Merged Mathieu Faverge requested to merge faverge/chameleon:feature/gemm_summa into master
2 files
+ 23
23
Compare changes
  • Side-by-side
  • Inline
Files
2
+ 21
0
@@ -68,6 +68,27 @@ int chameleon_desc_mat_free( CHAM_desc_t *desc )
return CHAMELEON_SUCCESS;
}
/**
* Internal function to return MPI rank of element A(m,n) with m,n = block indices
*/
int chameleon_getrankof_2d( const CHAM_desc_t *A, int m, int n )
{
int mm = m + A->i / A->mb;
int nn = n + A->j / A->nb;
return (mm % A->p) * A->q + (nn % A->q);
}
/**
* Internal function to return MPI rank of element DIAG(m,0) with m,n = block indices
*/
int chameleon_getrankof_2d_diag( const CHAM_desc_t *A, int m, int n )
{
int mm = m + A->i / A->mb;
assert( m == n );
return (mm % A->p) * A->q + (mm % A->q);
}
/**
******************************************************************************
*
Loading