Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
/**
*
* @file lass.h
*
* @brief LASs header definition.
*
* LASs is a software package provided by:
* Barcelona Supercomputing Center - Centro Nacional de Supercomputacion
*
* @author Pedro Valero-Lara pedro.valero@bsc.es
* @date 2017-09-07
* @reviewer
* @modified
*
**/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include <float.h>
#include <sys/time.h>
#include <math.h>
#include <time.h>
#include <unistd.h>
#if defined(LASs_WITH_MKL)
#include <mkl.h>
#include <mkl_lapacke.h>
#else
#include <cblas.h>
#include <lapacke.h>
#endif
#include "../mmsparse/mm_io.h"
#include "lass_macros.h"
#include "lass_types.h"
#ifndef DDSS_H
#define DDSS_H
// LASs-DDSs routines
// BLAS-3 routines
int ddss_dgemm( enum DDSS_TRANS TRANS_A, enum DDSS_TRANS TRANS_B,
int M, int N, int K,
double ALPHA, double *A, int LDA,
double *B, int LDB,
double BETA, double *C, int LDC );
int ddss_dsymm( enum DDSS_SIDE SIDE, enum DDSS_UPLO UPLO,
int M, int N,
double ALPHA, double *A, int LDA,
double *B, int LDB,
double BETA, double *C, int LDC );
int ddss_dtrsm( enum DDSS_SIDE SIDE, enum DDSS_UPLO UPLO,
enum DDSS_TRANS TRANS_A, enum DDSS_DIAG DIAG,
int M, int N,
const double ALPHA, double *A, int LDA,
double *B, int LDB );
int ddss_dtrmm( enum DDSS_SIDE SIDE, enum DDSS_UPLO UPLO,
enum DDSS_TRANS TRANS_A, enum DDSS_DIAG DIAG,
int M, int N,
const double ALPHA, double *A, int LDA,
double *B, int LDB );
int ddss_dsyrk( enum DDSS_UPLO UPLO, enum DDSS_TRANS TRANS_A,
int N, int K,
const double ALPHA, double *A, int LDA,
const double BETA, double *C, int LDC );
int ddss_dsyr2k( enum DDSS_UPLO UPLO, enum DDSS_TRANS TRANS,
int N, int K,
const double ALPHA, double *A, int LDA,
double *B, int LDB,
const double BETA, double *C, int LDC );
// LAPACK routines
int ddss_dnpgetrf( int M,
int N,
double *A, int LDA );
int ddss_dnpgesv( int N, int NRHS,
double *A, int LDA,
double *B, int LDB );
int ddss_dtpgetrf( int M,
int N,
double *A, int LDA,
int *ipiv );
int ddss_dppgetrf( int M,
int N,
double *A, int LDA,
int *ipiv );
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
int ddss_dtpgesv( int N, int NRHS,
double *A, int LDA,
int *IPIV,
double *B, int LDB );
int ddss_dpotrf( enum DDSS_UPLO UPLO,
int N,
double *A, int LDA );
int ddss_dposv( enum DDSS_UPLO UPLO,
int N, int NRHS,
double *A, int LDA,
double *B, int LDB );
// BLAS-3 kernels
enum LASS_RETURN kdgemm( enum DDSS_TRANS TRANS_A, enum DDSS_TRANS TRANS_B,
int M, int N, int K,
const double ALPHA, double *A, int LDA,
double *B, int LDB,
const double BETA, double *C, int LDC );
enum LASS_RETURN kdsymm( enum DDSS_SIDE SIDE, enum DDSS_UPLO UPLO,
int M, int N,
const double ALPHA, double *A, int LDA,
double *B, int LDB,
const double BETA, double *C, int LDC );
enum LASS_RETURN kdtrsm( enum DDSS_SIDE SIDE, enum DDSS_UPLO UPLO,
enum DDSS_TRANS TRANS_A, enum DDSS_DIAG DIAG,
int M, int N,
const double ALPHA, double *A, int LDA,
double *B, int LDB );
enum LASS_RETURN kdtrmm( enum DDSS_SIDE SIDE, enum DDSS_UPLO UPLO,
enum DDSS_TRANS TRANS_A, enum DDSS_DIAG DIAG,
int M, int N,
const double ALPHA, double *A, int LDA,
double *B, int LDB );
enum LASS_RETURN kdsyrk( enum DDSS_UPLO UPLO, enum DDSS_TRANS TRANS_A,
int N, int K,
const double ALPHA, double* A, int LDA,
const double BETA, double *C, int LDC );
enum LASS_RETURN kdsyr2k( enum DDSS_UPLO UPLO, enum DDSS_TRANS TRANS,
int N, int K,
const double ALPHA, double *A, int LDA,
double *B, int LDB,
const double BETA, double *C, int LDC );
// LAPACK kernels
enum LASS_RETURN kdnpgetrf( int M,
int N,
double *A, int LDA );
enum LASS_RETURN kdnpgesv( int N, int NRHS,
double *A, int LDA,
double *B, int LDB );
enum LASS_RETURN kdtpgetrf( int M,
int N,
double *A, int LDA,
int *ipiv );
enum LASS_RETURN kdppgetrf( int M,
int N,
double *A, int LDA,
int *ipiv );
enum LASS_RETURN kdtpgesv( int N, int NRHS,
double *A, int LDA,
int *IPIV,
double *B, int LDB );
enum LASS_RETURN kdpotrf( enum DDSS_UPLO UPLO,
int N,
double *A, int LDA );
enum LASS_RETURN kdposv( enum DDSS_UPLO UPLO,
int N, int NRHS,
double *A, int LDA,
double *B, int LDB );
// LAPACK codes
enum LASS_RETURN dnpgetrf( int M, int N, double *A, int LDA );
// ddss_tile.c routines
int ddss_tile_size( int M, int MT );
// row swap routines
void ddss_dtiled_laswp( int N, int mi, int ni, int MT, int NT,
double (*TILE_A)[NT][TILE_SIZE * TILE_SIZE],
int K1, int K2, int *ipiv, int incx );
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
// ddss_flat2tiled.c routines
void ddss_dflat2tiled( int M, int N, double *A, int LDA, int MT, int NT,
double (*TILE_A)[NT][TILE_SIZE * TILE_SIZE] );
void ddss_dsymtiled2flat( int M, int N, double *A, int LDA, int MT, int NT,
double (*TILE_A)[NT][TILE_SIZE * TILE_SIZE], enum DDSS_UPLO UPLO );
void ddss_dgather_tile( int M, int N, double *A, int LDA,
double *TILE_A, int MID, int NID );
// ddss_tiled2flat.c routines
void ddss_dtiled2flat( int M, int N, double *A, int LDA, int MT, int NT,
double (*TILE_A)[NT][TILE_SIZE * TILE_SIZE] );
void ddss_dtiled2flat_nb( int M, int N, double *A, int LDA, int MT, int NT,
double (*TILE_A)[NT][TILE_SIZE * TILE_SIZE] );
void ddss_dsymflat2tiled( int M, int N, double *A, int LDA, int MT, int NT,
double (*TILE_A)[NT][TILE_SIZE * TILE_SIZE], enum DDSS_UPLO UPLO );
void ddss_dsymtiled2flat_nb( int M, int N, double *A, int LDA, int MT, int NT,
double (*TILE_A)[NT][TILE_SIZE * TILE_SIZE], enum DDSS_UPLO UPLO );
void ddss_dscatter_tile( int M, int N, double *A, int LDA,
double *TILE_A, int MID, int NID );
#endif
#ifndef DSSS_H
#define DSSS_H
// LASs-DSSs routines
// DGTSV
int dsss_dgtsv( int N,
double *DL, double *D, double *DU,
double *RHS );
// DSPMV
int dsss_dspmv( int M, int N,
double ALPHA,
const double *VAL_A, const int *ROW_PTR_A, const int *COL_IND_A,
const double *X,
double BETA,
double *Y );
// DGTSV kernel
enum LASS_RETURN kdgtsv( int N,
double *DL, double *D, double *DU,
double *RHS );
// DSPMV kernel
enum LASS_RETURN kdspmv ( int M, int N,
double ALPHA,
const double *VAL_A, const int *ROW_PTR_A, const int *COL_IND_A,
const double *X,
double BETA,
double *Y );
// DGTSV code
void dgtsvseq( int N,
double *DL, double *D, double *DU,
double *RHS );
// DSPMV code
void dspmvseq( int M, int N,
double ALPHA,
const double *VAL_A, const int *ROW_PTR_A, const int *COL_IND_A,
const double *X,
double BETA,
double *Y );
#endif