-
Notifications
You must be signed in to change notification settings - Fork 0
/
openmp_blas.h
61 lines (47 loc) · 2.28 KB
/
openmp_blas.h
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
#include "common.h"
#include <omp.h>
//#ifndef OMPBLAS_H
//#define OMPBLAS_H
real_type openmp_dot (const int n, const real_type *v, const real_type *w);
void openmp_scal (const int n, const real_type alpha, real_type *v);
void openmp_axpy (const int n, const real_type alpha, const real_type *x, real_type *y);
void openmp_csr_matvec(const int n,
const int nnz,
const int *ia,
const int *ja,
const real_type *a,
const real_type *x,
real_type *result,
const real_type *al,
const real_type *bet);
void openmp_lower_triangular_solve(const int n,
const int nnz,
const int *lia,
const int *lja,
const real_type *la,
const real_type *diag,
const real_type *x,
real_type *result);
void openmp_upper_triangular_solve(const int n,
const int nnz,
const int *uia,
const int *uja,
const real_type *ua,
const real_type *diag,
const real_type *x,
real_type *result);
void openmp_vec_vec(const int n, const real_type *x, const real_type *y, real_type *res);
void openmp_vector_sqrt(const int n, const real_type *v, real_type *res);
void openmp_vector_reciprocal(const int n, const real_type *v, real_type *res);
void openmp_vec_copy(const int n, const real_type *src, real_type *dest);
void openmp_vec_zero(const int n, real_type *vec);
void initialize_ichol(const int n,
const int nnzA,
int *ia,
int *ja,
real_type *a,
int *lia,
int *lja,
real_type *la);
void openmp_ichol(const int *ia, const int *ja, real_type *a, int nnzA, pdata *prec_data, real_type *x, real_type *y);
//#endif