PaStiX Handbook 6.4.0
Loading...
Searching...
No Matches
pastix_sstarpu.h
Go to the documentation of this file.
1/**
2 *
3 * @file pastix_sstarpu.h
4 *
5 * Pastix StarPU codelets header
6 *
7 * @copyright 2016-2024 Bordeaux INP, CNRS (LaBRI UMR 5800), Inria,
8 * Univ. Bordeaux. All rights reserved.
9 *
10 * @version 6.4.0
11 * @author Mathieu Faverge
12 * @author Pierre Ramet
13 * @author Vincent Bridonneau
14 * @author Alycia Lisito
15 * @date 2024-07-05
16 *
17 * @generated from /builds/2mk6rsew/0/solverstack/pastix/sopalin/starpu/pastix_zstarpu.h, normal z -> s, Tue Feb 25 14:34:34 2025
18 *
19 **/
20#ifndef _pastix_sstarpu_h_
21#define _pastix_sstarpu_h_
22
23void starpu_task_cblk_sgetrfsp( sopalin_data_t *sopalin_data,
24 SolverCblk *cblk,
25 int prio );
26void starpu_task_cblk_ssytrfsp( sopalin_data_t *sopalin_data,
27 SolverCblk *cblk,
28 int prio );
29void starpu_task_cblk_spotrfsp( sopalin_data_t *sopalin_data,
30 SolverCblk *cblk,
31 int prio );
32void starpu_task_cblk_spotrfsp( sopalin_data_t *sopalin_data,
33 SolverCblk *cblk,
34 int prio );
35void starpu_task_cblk_ssytrfsp( sopalin_data_t *sopalin_data,
36 SolverCblk *cblk,
37 int prio );
38
39void starpu_task_blok_sgetrf( sopalin_data_t *sopalin_data,
40 SolverCblk *cblk,
41 int prio );
42void starpu_task_blok_ssytrf( sopalin_data_t *sopalin_data,
43 SolverCblk *cblk,
44 int prio );
45void starpu_task_blok_spotrf( sopalin_data_t *sopalin_data,
46 SolverCblk *cblk,
47 int prio );
48void starpu_task_blok_spotrf( sopalin_data_t *sopalin_data,
49 SolverCblk *cblk,
50 int prio );
51void starpu_task_blok_ssytrf( sopalin_data_t *sopalin_data,
52 SolverCblk *cblk,
53 int prio );
54void starpu_stask_cblk_sdiag( sopalin_data_t *sopalin_data,
55 pastix_rhs_t rhsb,
56 SolverCblk *cblk,
57 int prio );
58void starpu_stask_blok_sgemm( sopalin_data_t *sopalin_data,
59 pastix_rhs_t rhsb,
60 pastix_coefside_t coef,
61 pastix_side_t side,
62 pastix_trans_t trans,
63 const SolverCblk *cblk,
64 const SolverBlok *blok,
65 SolverCblk *fcbk,
66 pastix_int_t prio );
67void starpu_stask_blok_strsm( sopalin_data_t *sopalin_data,
68 pastix_rhs_t rhsb,
69 pastix_coefside_t coef,
70 pastix_side_t side,
71 pastix_uplo_t uplo,
72 pastix_trans_t trans,
73 pastix_diag_t diag,
74 const SolverCblk *cblk,
75 pastix_int_t prio );
76
77void starpu_task_cblk_sgemmsp( sopalin_data_t *sopalin_data,
78 pastix_coefside_t sideA,
79 pastix_coefside_t sideB,
80 pastix_trans_t trans,
81 const SolverCblk *cblk,
82 const SolverBlok *blok,
83 SolverCblk *fcblk,
84 int prio );
85void starpu_task_cblk_sadd_recv( sopalin_data_t *sopalin_data,
86 pastix_coefside_t side,
87 const SolverCblk *cblk,
88 SolverCblk *fcblk,
89 int prio );
90void starpu_task_cblk_sadd_fanin( sopalin_data_t *sopalin_data,
91 pastix_coefside_t side,
92 const SolverCblk *cblk,
93 int prio );
94void starpu_task_blok_sadd_recv( sopalin_data_t *sopalin_data,
95 pastix_coefside_t side,
96 const SolverCblk *cblk,
97 const SolverBlok *blok,
98 SolverCblk *fcblk,
99 SolverBlok *fblok,
100 int prio );
101void starpu_task_blok_sadd_fanin( sopalin_data_t *sopalin_data,
102 pastix_coefside_t side,
103 const SolverCblk *cblk,
104 const SolverBlok *blok,
105 int prio );
106void starpu_task_blok_sgemmsp( sopalin_data_t *sopalin_data,
107 pastix_coefside_t sideA,
108 pastix_coefside_t sideB,
109 pastix_trans_t trans,
110 SolverCblk *cblk,
111 SolverCblk *fcblk,
112 const SolverBlok *blokA,
113 const SolverBlok *blokB,
114 int prio );
115
116void starpu_task_blok_strsmsp( sopalin_data_t *sopalin_data,
117 pastix_coefside_t coef,
118 pastix_side_t side,
119 pastix_uplo_t uplo,
120 pastix_trans_t trans,
121 pastix_diag_t diag,
122 const SolverCblk *cblk,
123 SolverBlok *blok,
124 int prio );
125
126void starpu_task_blok_sscalo( sopalin_data_t *sopalin_data,
127 pastix_trans_t trans,
128 const SolverCblk *cblk,
129 SolverBlok *blok,
130 int prio );
131
132void starpu_task_sadd_1dp_fanin( sopalin_data_t *sopalin_data,
133 pastix_coefside_t side,
134 const SolverCblk *cblk,
135 int prio );
136void starpu_task_sadd_1dp_recv( sopalin_data_t *sopalin_data,
137 pastix_coefside_t side,
138 const SolverCblk *cblk,
139 int prio );
140void starpu_task_sadd_2d_fanin( sopalin_data_t *sopalin_data,
141 pastix_coefside_t side,
142 const SolverCblk *cblk,
143 int prio );
144void starpu_task_sadd_2d_recv( sopalin_data_t *sopalin_data,
145 pastix_coefside_t side,
146 const SolverCblk *cblk,
147 int prio );
148void starpu_task_sadd_fanin( sopalin_data_t *sopalin_data,
149 pastix_coefside_t side,
150 const SolverCblk *cblk,
151 int prio );
152void starpu_task_sadd_recv( sopalin_data_t *sopalin_data,
153 pastix_coefside_t side,
154 const SolverCblk *cblk,
155 int prio );
156void starpu_stask_blok_sadd_fwd_recv( sopalin_data_t *sopalin_data,
157 pastix_rhs_t rhsb,
158 const SolverCblk *cblk,
159 SolverCblk *fcblk,
160 int prio );
161
162void starpu_stask_blok_scpy_bwd_recv( sopalin_data_t *sopalin_data,
163 pastix_rhs_t rhsb,
164 SolverCblk *cblk,
165 const SolverCblk *fcblk,
166 int prio );
167
168void starpu_sdiag ( pastix_data_t *pastix_data,
169 sopalin_data_t *sopalin_data,
170 pastix_rhs_t rhsb );
171void starpu_spotrf( pastix_data_t *pastix_data,
172 sopalin_data_t *sopalin_data );
173void starpu_spotrf( pastix_data_t *pastix_data,
174 sopalin_data_t *sopalin_data );
175void starpu_sgetrf( pastix_data_t *pastix_data,
176 sopalin_data_t *sopalin_data );
177void starpu_ssytrf( pastix_data_t *pastix_data,
178 sopalin_data_t *sopalin_data );
179void starpu_ssytrf( pastix_data_t *pastix_data,
180 sopalin_data_t *sopalin_data );
181void starpu_strsm ( pastix_data_t *pastix_data,
182 const args_solve_t *enums,
183 sopalin_data_t *sopalin_data,
184 pastix_rhs_t rhsb );
185
186#endif /* _pastix_sstarpu_h_ */
pastix_int_t
BEGIN_C_DECLS typedef int pastix_int_t
Definition datatypes.h:51
pastix_diag_t
enum pastix_diag_e pastix_diag_t
Diagonal.
pastix_uplo_t
enum pastix_uplo_e pastix_uplo_t
Upper/Lower part.
pastix_side_t
enum pastix_side_e pastix_side_t
Side of the operation.
pastix_trans_t
enum pastix_trans_e pastix_trans_t
Transpostion.
pastix_coefside_t
enum pastix_coefside_e pastix_coefside_t
Data blocks used in the kernel.
starpu_task_blok_sadd_fanin
void starpu_task_blok_sadd_fanin(sopalin_data_t *sopalin_data, pastix_coefside_t side, const SolverCblk *cblk, const SolverBlok *blok, int prio)
Insert the task to add a fanin cblk on the emitter side. Note that this task is submitted only to emi...
Definition codelet_blok_sadd.c:380
starpu_task_cblk_spotrfsp
void starpu_task_cblk_spotrfsp(sopalin_data_t *sopalin_data, SolverCblk *cblk, int prio)
TODO.
Definition codelet_cblk_spotrfsp.c:223
starpu_stask_blok_scpy_bwd_recv
void starpu_stask_blok_scpy_bwd_recv(sopalin_data_t *sopalin_data, pastix_rhs_t rhsb, SolverCblk *cblk, const SolverCblk *fcblk, int prio)
Insert the task to add a fanin cblk on the receiver side (The fanin is seen on this side as the RECV ...
Definition codelet_solve_scpy.c:135
starpu_task_blok_sscalo
void starpu_task_blok_sscalo(sopalin_data_t *sopalin_data, pastix_trans_t trans, const SolverCblk *cblk, SolverBlok *blok, int prio)
TODO.
Definition codelet_blok_sscalo.c:171
starpu_task_blok_spotrf
void starpu_task_blok_spotrf(sopalin_data_t *sopalin_data, SolverCblk *cblk, int prio)
TODO.
Definition codelet_blok_spotrfsp.c:220
starpu_task_blok_sadd_recv
void starpu_task_blok_sadd_recv(sopalin_data_t *sopalin_data, pastix_coefside_t side, const SolverCblk *cblk, const SolverBlok *blok, SolverCblk *fcblk, SolverBlok *fblok, int prio)
Insert the task to add a fanin cblk on the receiver side (The fanin is seen on this side as the RECV ...
Definition codelet_blok_sadd.c:288
starpu_task_blok_sgemmsp
void starpu_task_blok_sgemmsp(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.
Definition codelet_blok_sgemmsp.c:287
starpu_task_blok_ssytrf
void starpu_task_blok_ssytrf(sopalin_data_t *sopalin_data, SolverCblk *cblk, int prio)
TODO.
Definition codelet_blok_ssytrfsp.c:218
starpu_task_cblk_sgetrfsp
void starpu_task_cblk_sgetrfsp(sopalin_data_t *sopalin_data, SolverCblk *cblk, int prio)
TODO.
Definition codelet_cblk_sgetrfsp.c:225
starpu_task_cblk_sadd_fanin
void starpu_task_cblk_sadd_fanin(sopalin_data_t *sopalin_data, pastix_coefside_t side, const SolverCblk *cblk, int prio)
Insert the task to add a fanin cblk on the emitter side. Note that this task is submitted only to emi...
Definition codelet_cblk_sadd.c:358
starpu_task_cblk_sadd_recv
void starpu_task_cblk_sadd_recv(sopalin_data_t *sopalin_data, pastix_coefside_t side, const SolverCblk *cblk, SolverCblk *fcblk, int prio)
Insert the task to add a fanin cblk on the receiver side (The fanin is seen on this side as the RECV ...
Definition codelet_cblk_sadd.c:275
starpu_task_blok_strsmsp
void starpu_task_blok_strsmsp(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.
Definition codelet_blok_strsmsp.c:277
starpu_task_cblk_ssytrfsp
void starpu_task_cblk_ssytrfsp(sopalin_data_t *sopalin_data, SolverCblk *cblk, int prio)
TODO.
Definition codelet_cblk_ssytrfsp.c:231
starpu_stask_blok_sgemm
void starpu_stask_blok_sgemm(sopalin_data_t *sopalin_data, pastix_rhs_t rhsb, pastix_coefside_t coef, pastix_side_t side, pastix_trans_t trans, const SolverCblk *cblk, const SolverBlok *blok, SolverCblk *fcbk, pastix_int_t prio)
Submit a task to perform a gemm.
Definition codelet_solve_sgemm.c:157
starpu_stask_blok_sadd_fwd_recv
void starpu_stask_blok_sadd_fwd_recv(sopalin_data_t *sopalin_data, pastix_rhs_t rhsb, const SolverCblk *cblk, SolverCblk *fcblk, int prio)
Insert the task to add a fanin cblk on the receiver side (The fanin is seen on this side as the RECV ...
Definition codelet_solve_sadd.c:128
starpu_task_cblk_sgemmsp
void starpu_task_cblk_sgemmsp(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.
Definition codelet_cblk_sgemmsp.c:285
starpu_task_blok_sgetrf
void starpu_task_blok_sgetrf(sopalin_data_t *sopalin_data, SolverCblk *cblk, int prio)
TODO.
Definition codelet_blok_sgetrfsp.c:222
pastix_data_s
Main PaStiX data structure.
Definition pastixdata.h:68
pastix_rhs_s
Main PaStiX RHS structure.
Definition pastixdata.h:155
starpu_stask_cblk_sdiag
void starpu_stask_cblk_sdiag(sopalin_data_t *sopalin_data, pastix_rhs_t rhsb, SolverCblk *cblk, int prio)
Submit a task to perform a diagonal solve related to one cblk to all the right hand side.
Definition codelet_solve_sdiag.c:112
starpu_sdiag
void starpu_sdiag(pastix_data_t *pastix_data, sopalin_data_t *sopalin_data, pastix_rhs_t rhsb)
Apply the diagonal solve (StarPU version)
Definition starpu_sdiag.c:88
starpu_sgetrf
void starpu_sgetrf(pastix_data_t *pastix_data, sopalin_data_t *sopalin_data)
Perform a sparse LU factorization using StarPU runtime.
Definition starpu_sgetrf.c:587
starpu_spotrf
void starpu_spotrf(pastix_data_t *pastix_data, sopalin_data_t *sopalin_data)
Perform a sparse Cholesky factorization using StarPU runtime.
Definition starpu_spotrf.c:533
starpu_ssytrf
void starpu_ssytrf(pastix_data_t *pastix_data, sopalin_data_t *sopalin_data)
Perform a sparse LDL^t factorization using StarPU runtime.
Definition starpu_ssytrf.c:570
starpu_stask_blok_strsm
void starpu_stask_blok_strsm(sopalin_data_t *sopalin_data, pastix_rhs_t rhsb, pastix_coefside_t coef, pastix_side_t side, pastix_uplo_t uplo, pastix_trans_t trans, pastix_diag_t diag, const SolverCblk *cblk, pastix_int_t prio)
Submit a task to do a trsm related to a diagonal block of the matrix A.
Definition codelet_solve_strsm.c:133
starpu_strsm
void starpu_strsm(pastix_data_t *pastix_data, const args_solve_t *enums, sopalin_data_t *sopalin_data, pastix_rhs_t rhsb)
Apply the TRSM solve (StarPU version).
Definition starpu_strsm.c:365
starpu_task_sadd_1dp_recv
void starpu_task_sadd_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.
Definition starpu_task_sadd.c:77
starpu_task_sadd_1dp_fanin
void starpu_task_sadd_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.
Definition starpu_task_sadd.c:46
starpu_task_sadd_fanin
void starpu_task_sadd_fanin(sopalin_data_t *sopalin_data, pastix_coefside_t side, const SolverCblk *cblk, int prio)
Submits starpu zadd task to send the fanin cblk.
Definition starpu_task_sadd.c:241
starpu_task_sadd_recv
void starpu_task_sadd_recv(sopalin_data_t *sopalin_data, pastix_coefside_t side, const SolverCblk *cblk, int prio)
Submits starpu zadd task to send the recv cblk.
Definition starpu_task_sadd.c:205
starpu_task_sadd_2d_fanin
void starpu_task_sadd_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.
Definition starpu_task_sadd.c:109
starpu_task_sadd_2d_recv
void starpu_task_sadd_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.
Definition starpu_task_sadd.c:154
args_solve_s
Arguments for the solve.
Definition solver.h:88
solver_blok_s
Solver block structure.
Definition solver.h:141
solver_cblk_s
Solver column block structure.
Definition solver.h:161