QMCPACK
hipBLAS.hpp
Go to the documentation of this file.
1 //////////////////////////////////////////////////////////////////////////////////////
2 // This file is distributed under the University of Illinois/NCSA Open Source License.
3 // See LICENSE file in top directory for details.
4 //
5 // Copyright (c) 2021 QMCPACK developers.
6 // Copyright(C) 2021 Advanced Micro Devices, Inc. All rights reserved.
7 //
8 // File developed by: Jakub Kurzak, jakurzak@amd.com, Advanced Micro Devices, Inc.
9 // Ye Luo, yeluo@anl.gov, Argonne National Laboratory
10 //
11 // File created by: Jakub Kurzak, jakurzak@amd.com, Advanced Micro Devices, Inc.
12 //////////////////////////////////////////////////////////////////////////////////////
13 
14 
15 #ifndef HIPBLAS_HPP
16 #define HIPBLAS_HPP
17 
18 #include <hipblas/hipblas.h>
19 #include <hip/hip_complex.h>
20 
21 //------------------------------------------------------------------------------
22 hipblasStatus_t
23 hipblasCgemmBatched(hipblasHandle_t handle,
24  hipblasOperation_t transa,
25  hipblasOperation_t transb,
26  int m,
27  int n,
28  int k,
29  const hipComplex *alpha,
30  const hipComplex *const Aarray[],
31  int lda,
32  const hipComplex *const Barray[],
33  int ldb,
34  const hipComplex *beta,
35  hipComplex *const Carray[],
36  int ldc,
37  int batchCount);
38 
39 hipblasStatus_t
40 hipblasZgemmBatched(hipblasHandle_t handle,
41  hipblasOperation_t transa,
42  hipblasOperation_t transb,
43  int m,
44  int n,
45  int k,
46  const hipDoubleComplex *alpha,
47  const hipDoubleComplex *const Aarray[],
48  int lda,
49  const hipDoubleComplex *const Barray[],
50  int ldb,
51  const hipDoubleComplex *beta,
52  hipDoubleComplex *const Carray[],
53  int ldc,
54  int batchCount);
55 
56 //------------------------------------------------------------------------------
57 hipblasStatus_t
58 hipblasSgetrfBatched_(hipblasHandle_t handle,
59  int n,
60  float *const A[],
61  int lda,
62  int *P,
63  int *info,
64  int batchSize);
65 
66 hipblasStatus_t
67 hipblasDgetrfBatched_(hipblasHandle_t handle,
68  int n,
69  double *const A[],
70  int lda,
71  int *P,
72  int *info,
73  int batchSize);
74 
75 hipblasStatus_t
76 hipblasCgetrfBatched_(hipblasHandle_t handle,
77  int n,
78  hipComplex *const A[],
79  int lda,
80  int *P,
81  int *info,
82  int batchSize);
83 
84 hipblasStatus_t
85 hipblasZgetrfBatched_(hipblasHandle_t handle,
86  int n,
87  hipDoubleComplex *const A[],
88  int lda,
89  int *P,
90  int *info,
91  int batchSize);
92 
93 //------------------------------------------------------------------------------
94 hipblasStatus_t
95 hipblasSgetriBatched_(hipblasHandle_t handle,
96  int n,
97  const float *const A[],
98  int lda,
99  const int *P,
100  float *const C[],
101  int ldc,
102  int *info,
103  int batchSize);
104 
105 hipblasStatus_t
106 hipblasDgetriBatched_(hipblasHandle_t handle,
107  int n,
108  const double *const A[],
109  int lda,
110  const int *P,
111  double *const C[],
112  int ldc,
113  int *info,
114  int batchSize);
115 
116 hipblasStatus_t
117 hipblasCgetriBatched_(hipblasHandle_t handle,
118  int n,
119  const hipComplex *const A[],
120  int lda,
121  const int *P,
122  hipComplex *const C[],
123  int ldc,
124  int *info,
125  int batchSize);
126 
127 hipblasStatus_t
128 hipblasZgetriBatched_(hipblasHandle_t handle,
129  int n,
130  const hipDoubleComplex *const A[],
131  int lda,
132  const int *P,
133  hipDoubleComplex *const C[],
134  int ldc,
135  int *info,
136  int batchSize);
137 
138 #endif /* HIPBLAS_HPP */
hipblasSgetriBatched_
hipblasStatus_t hipblasSgetriBatched_(hipblasHandle_t handle, int n, const float *const A[], int lda, const int *P, float *const C[], int ldc, int *info, int batchSize)
Definition: hipBLAS.cpp:129
hipblasDgetriBatched_
hipblasStatus_t hipblasDgetriBatched_(hipblasHandle_t handle, int n, const double *const A[], int lda, const int *P, double *const C[], int ldc, int *info, int batchSize)
Definition: hipBLAS.cpp:144
hipblasCgemmBatched
hipblasStatus_t hipblasCgemmBatched(hipblasHandle_t handle, hipblasOperation_t transa, hipblasOperation_t transb, int m, int n, int k, const hipComplex *alpha, const hipComplex *const Aarray[], int lda, const hipComplex *const Barray[], int ldb, const hipComplex *beta, hipComplex *const Carray[], int ldc, int batchCount)
Definition: hipBLAS.cpp:20
hipblasCgetriBatched_
hipblasStatus_t hipblasCgetriBatched_(hipblasHandle_t handle, int n, const hipComplex *const A[], int lda, const int *P, hipComplex *const C[], int ldc, int *info, int batchSize)
Definition: hipBLAS.cpp:159
hipblasZgemmBatched
hipblasStatus_t hipblasZgemmBatched(hipblasHandle_t handle, hipblasOperation_t transa, hipblasOperation_t transb, int m, int n, int k, const hipDoubleComplex *alpha, const hipDoubleComplex *const Aarray[], int lda, const hipDoubleComplex *const Barray[], int ldb, const hipDoubleComplex *beta, hipDoubleComplex *const Carray[], int ldc, int batchCount)
Definition: hipBLAS.cpp:41
hipblasZgetriBatched_
hipblasStatus_t hipblasZgetriBatched_(hipblasHandle_t handle, int n, const hipDoubleComplex *const A[], int lda, const int *P, hipDoubleComplex *const C[], int ldc, int *info, int batchSize)
Definition: hipBLAS.cpp:175
hipblasCgetrfBatched_
hipblasStatus_t hipblasCgetrfBatched_(hipblasHandle_t handle, int n, hipComplex *const A[], int lda, int *P, int *info, int batchSize)
Definition: hipBLAS.cpp:96
hipblasDgetrfBatched_
hipblasStatus_t hipblasDgetrfBatched_(hipblasHandle_t handle, int n, double *const A[], int lda, int *P, int *info, int batchSize)
Definition: hipBLAS.cpp:80
hipblasSgetrfBatched_
hipblasStatus_t hipblasSgetrfBatched_(hipblasHandle_t handle, int n, float *const A[], int lda, int *P, int *info, int batchSize)
Definition: hipBLAS.cpp:64
hipblasZgetrfBatched_
hipblasStatus_t hipblasZgetrfBatched_(hipblasHandle_t handle, int n, hipDoubleComplex *const A[], int lda, int *P, int *info, int batchSize)
Definition: hipBLAS.cpp:112