22 #include "sopalin/sopalin_data.h"
52 if ( cblk->
cblktype & CBLK_TASKS_2D ) {
56 for ( blok = cblk->
fblokptr + 1, m = 0; blok < lblk; blok++, m++ ) {
63 while ( ( blok < lblk ) &&
64 ( blok[0].fcblknm == blok[1].fcblknm ) &&
65 ( blok[0].lcblknm == blok[1].lcblknm ) )
108 if ( cblk->
cblktype & CBLK_TASKS_2D ) {
110 for ( blokA = blokB; blokA < lblk; blokA++ ) {
113 cblk, fcblk, blokA, blokB, prio );
116 while ( ( blokA < lblk ) &&
117 ( blokA[0].fcblknm == blokA[1].fcblknm ) &&
118 ( blokA[0].lcblknm == blokA[1].lcblknm ) )
126 cblk, blokB, fcblk, prio );
163 for ( k = 0; k < solvmtx->
cblknbr; k++, cblk++ ) {
165 if ( cblk->
cblktype & CBLK_IN_SCHUR ) {
172 if ( cblk->
cblktype & CBLK_FANIN ) {
191 for ( m = 0; blok < lblk; blok++, m++ ) {
193 cblk_n = fcblk - solvmtx->
cblktab;
197 cblknbr - pastix_imin( k + m, cblk_n ) );
237 for ( k = 0; k < solvmtx->
cblknbr; k++, fcblk++ ) {
241 for ( m = fcblk[0].brownum; m < fcblk[1].
brownum; m++ ) {
245 if ( cblk->
cblktype & CBLK_IN_SCHUR ) {
248 assert( !( cblk->
cblktype & CBLK_FANIN ) );
258 cblk, blok, fcblk, prio );
261 if ( fcblk->
cblktype & ( CBLK_IN_SCHUR | CBLK_RECV ) ) {
266 if ( fcblk->
cblktype & CBLK_FANIN ) {
276 for ( k = 0; k < solvmtx->
cblknbr; k++, cblk++ ) {
277 if ( !(cblk->
cblktype & (CBLK_FANIN|CBLK_RECV)) ) {
319 for ( k = 0; k <= solvmtx->
cblkmax1d; k++, cblk++ ) {
321 if ( cblk->
cblktype & CBLK_IN_SCHUR ) {
325 if ( cblk->
cblktype & CBLK_TASKS_2D ) {
332 if ( cblk->
cblktype & CBLK_FANIN ) {
351 for ( m = 0; blok < lblk; blok++, m++ ) {
353 cblk_n = fcblk - solvmtx->
cblktab;
356 cblknbr - pastix_imin( k + m, cblk_n ) );
363 for (k=solvmtx->
cblkmin2d; k<solvmtx->cblknbr; k++, cblk++ ){
365 if ( !(cblk->
cblktype & CBLK_TASKS_2D) ) {
369 if ( cblk->
cblktype & CBLK_IN_SCHUR ) {
376 if ( cblk->
cblktype & CBLK_FANIN ) {
393 for ( blok = cblk->
fblokptr + 1, m = 0; blok < lblk; blok++, m++ ) {
399 cblknbr - pastix_imin( k + m, cblk_n ) );
402 while( ( blok < lblk ) &&
403 ( blok[0].fcblknm == blok[1].fcblknm ) &&
404 ( blok[0].lcblknm == blok[1].lcblknm ) )
448 for ( k = 0; k < cblknbr; k++, fcblk++ ) {
452 for ( m = fcblk[0].brownum; m < fcblk[1].
brownum; m++ ) {
457 if ( cblk->
cblktype & CBLK_IN_SCHUR ) {
460 assert( !(cblk->
cblktype & CBLK_FANIN ) );
462 if( ( cblk->
cblktype & CBLK_TASKS_2D ) &&
463 ( blok_prev != NULL ) &&
480 if ( fcblk->
cblktype & ( CBLK_IN_SCHUR | CBLK_RECV ) ) {
484 if ( fcblk->
cblktype & CBLK_FANIN ) {
494 for ( k = 0; k < solvmtx->
cblknbr; k++, cblk++ ) {
495 if ( !(cblk->
cblktype & (CBLK_FANIN|CBLK_RECV)) ) {
531 sopalin_data_t *sopalin_data )
540 if (pastix_data->
starpu == NULL) {
545 if ( sdesc == NULL ) {
552 sdesc = sopalin_data->
solvmtx->starpu_desc;
555 starpu_profiling_status_set(STARPU_PROFILING_ENABLE);
556 #if defined(STARPU_USE_FXT)
558 starpu_fxt_start_profiling();
561 #if defined(PASTIX_STARPU_STATS)
589 #if defined(PASTIX_STARPU_STATS)
594 starpu_task_wait_for_all();
595 #if defined(PASTIX_WITH_MPI)
596 starpu_mpi_wait_for_all( pastix_data->
pastix_comm );
600 #if defined(STARPU_USE_FXT)
602 starpu_fxt_stop_profiling();
605 starpu_profiling_status_set(STARPU_PROFILING_DISABLE);
606 #if defined(PASTIX_STARPU_STATS)
608 print_stats( sub, com, pastix_data->
solvmatr );
BEGIN_C_DECLS typedef int pastix_int_t
void starpu_sparse_matrix_getoncpu(starpu_sparse_matrix_desc_t *desc)
Submit asynchronous calls to retrieve the data on main memory.
void starpu_task_blok_cpxtrf(sopalin_data_t *sopalin_data, SolverCblk *cblk, int prio)
TODO.
void starpu_task_cblk_cpxtrfsp(sopalin_data_t *sopalin_data, SolverCblk *cblk, int prio)
TODO.
void starpu_sparse_matrix_init(SolverMatrix *solvmtx, pastix_mtxtype_t mtxtype, int nodes, int myrank, pastix_coeftype_t flttype)
Generate the StarPU descriptor of the sparse matrix.
void pastix_starpu_init(pastix_data_t *pastix, int *argc, char **argv[], const int *bindtab)
Startup the StarPU runtime system.
void starpu_task_blok_ctrsmsp(sopalin_data_t *sopalin_data, pastix_coefside_t coef, pastix_side_t side, pastix_uplo_t uplo, pastix_trans_t trans, pastix_diag_t diag, const SolverCblk *cblk, SolverBlok *blok, int prio)
StarPU GPU implementation.
void starpu_task_cblk_cgemmsp(sopalin_data_t *sopalin_data, pastix_coefside_t sideA, pastix_coefside_t sideB, pastix_trans_t trans, const SolverCblk *cblk, const SolverBlok *blok, SolverCblk *fcblk, int prio)
StarPU GPU implementation.
void starpu_sparse_cblk_wont_use(pastix_coefside_t coef, SolverCblk *cblk)
Submit asynchronous calls to retrieve the data on main memory.
void starpu_task_blok_cgemmsp(sopalin_data_t *sopalin_data, pastix_coefside_t sideA, pastix_coefside_t sideB, pastix_trans_t trans, SolverCblk *cblk, SolverCblk *fcblk, const SolverBlok *blokA, const SolverBlok *blokB, int prio)
StarPU GPU implementation.
StarPU descriptor stucture for the sparse matrix.
PASTIX_Comm inter_node_comm
Main PaStiX data structure.
void starpu_task_pxtrf_cgemmsp(sopalin_data_t *sopalin_data, SolverCblk *cblk, const SolverBlok *blokB, SolverCblk *fcblk, int prio)
Submits starpu cgemmsp cblk or blok task.
void starpu_cpxtrf_sp1dplus_ll(sopalin_data_t *sopalin_data, starpu_sparse_matrix_desc_t *desc)
Perform a sparse LL^t factorization with 1D kernels.
void starpu_cpxtrf_sp2d_rl(sopalin_data_t *sopalin_data, starpu_sparse_matrix_desc_t *desc)
Perform a sparse LL^t factorization with 1D and 2D kernels.
void starpu_cpxtrf_sp2d_ll(sopalin_data_t *sopalin_data, starpu_sparse_matrix_desc_t *desc)
Perform a sparse LL^t factorization with 1D and 2D kernels.
void starpu_cpxtrf_sp1dplus_rl(sopalin_data_t *sopalin_data, starpu_sparse_matrix_desc_t *desc)
Perform a sparse LL^t factorization with 1D kernels.
void starpu_cpxtrf(pastix_data_t *pastix_data, sopalin_data_t *sopalin_data)
Perform a sparse LL^t factorization using StarPU runtime.
void starpu_task_cpxtrfsp(sopalin_data_t *sopalin_data, SolverCblk *cblk, int prio)
Submits starpu pxtrfsp cblk or blok task.
void starpu_task_cadd_recv(sopalin_data_t *sopalin_data, pastix_coefside_t side, const SolverCblk *cblk, int prio)
Submits starpu zadd task to send the recv cblk.
void starpu_task_cadd_1dp_recv(sopalin_data_t *sopalin_data, pastix_coefside_t side, const SolverCblk *cblk, int prio)
Submits starpu zadd task to receive and add the recv cblk.
void starpu_task_cadd_fanin(sopalin_data_t *sopalin_data, pastix_coefside_t side, const SolverCblk *cblk, int prio)
Submits starpu zadd task to send the fanin cblk.
void starpu_task_cadd_2d_fanin(sopalin_data_t *sopalin_data, pastix_coefside_t side, const SolverCblk *cblk, int prio)
Submits starpu zadd task to send the fanin block.
void starpu_task_cadd_1dp_fanin(sopalin_data_t *sopalin_data, pastix_coefside_t side, const SolverCblk *cblk, int prio)
Submits starpu zadd task to send the fanin cblk.
void starpu_task_cadd_2d_recv(sopalin_data_t *sopalin_data, pastix_coefside_t side, const SolverCblk *cblk, int prio)
Submits starpu zadd task to receive and add the recv block.
SolverBlok *restrict bloktab
pastix_int_t *restrict browtab
SolverCblk *restrict cblktab
Solver column block structure.
Solver column block structure.