-
Notifications
You must be signed in to change notification settings - Fork 0
/
hip_blas.h
86 lines (69 loc) · 3.08 KB
/
hip_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
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
#include "common.h"
#ifndef HIPBLAS_H
#define HIPBLAS_H
void initialize_handles();
void analyze_spmv(const int n,
const int nnz,
int *ia,
int *ja,
real_type *a,
const real_type *x,
real_type *result,
char * option);
void initialize_and_analyze_L_and_U_solve(const int n,
const int nnzL,
int *lia,
int *lja,
real_type *la,
const int nnzU,
int *uia,
int *uja,
real_type *ua);
real_type hip_dot (const int n, const real_type *v, const real_type *w);
void hip_scal (const int n, const real_type alpha, real_type *v);
void hip_axpy (const int n, const real_type alpha, const real_type *x, real_type *y);
void hip_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,
const char *kind);
void hip_lower_triangular_solve(const int n,
const int nnzL,
const int *lia,
const int *lja,
const real_type *la,
const real_type *diagonal,
const real_type *x,
real_type *result);
void hip_upper_triangular_solve(const int n,
const int nnzU,
const int *uia,
const int *uja,
const real_type *ua,
const real_type *diagonal,
const real_type *x,
real_type *result);
void hip_vec_vec(const int n, const real_type *x, const real_type *y, real_type *res);
void hip_vector_reciprocal(const int n, const real_type *v, real_type *res);
void hip_vector_sqrt(const int n, const real_type *v, real_type *res);
void hip_vec_copy(const int n, const real_type *src, real_type *dest);
void hip_vec_zero(const int n, real_type *vec);
void hip_vec_set(const int n, real_type value, real_type *vec);
void initialize_ichol(const int n,
const int nnzA,
int *ia,
int *ja,
real_type *a);
void hip_ichol(const int *ia,
const int *ja,
real_type *a,
const int nnzA,
pdata *prec_data,
real_type *x,
real_type *y);
#endif