File: slepccupmblas.h.html

package info (click to toggle)
slepc 3.24.1%2Bdfsg1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 122,028 kB
  • sloc: ansic: 104,353; javascript: 12,732; python: 5,958; f90: 3,312; cpp: 1,528; makefile: 761; xml: 679; sh: 347
file content (86 lines) | stat: -rw-r--r-- 7,142 bytes parent folder | download 
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
 
<center><a href="https://gitlab.com/slepc/slepc/-/blob/bbccb785e5ebe6829936989c71776839602f087b/include/slepccupmblas.h">Actual source code: slepccupmblas.h</a></center><br>

<html>
<head>
<title></title>
<meta name="generator" content="c2html 0.9.6">
<meta name="date" content="2025-11-07T08:27:34+00:00">
</head>

<body bgcolor="#FFFFFF">
<pre width=80>
<a name="line1">  1: </a><font color="#B22222">/*</font>
<a name="line2">  2: </a><font color="#B22222">   - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -</font>
<a name="line3">  3: </a><font color="#B22222">   SLEPc - Scalable Library for Eigenvalue Problem Computations</font>
<a name="line4">  4: </a><font color="#B22222">   Copyright (c) 2002-, Universitat Politecnica de Valencia, Spain</font>

<a name="line6">  6: </a><font color="#B22222">   This file is part of SLEPc.</font>
<a name="line7">  7: </a><font color="#B22222">   SLEPc is distributed under a 2-clause BSD license (see LICENSE).</font>
<a name="line8">  8: </a><font color="#B22222">   - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -</font>
<a name="line9">  9: </a><font color="#B22222">*/</font>
<a name="line10"> 10: </a><font color="#B22222">/*</font>
<a name="line11"> 11: </a><font color="#B22222">   Macro definitions to use cuBLAS and hipBLAS functionality</font>
<a name="line12"> 12: </a><font color="#B22222">*/</font>

<a name="line14"> 14: </a><font color="#A020F0">#pragma once</font>

<a name="line16"> 16: </a><font color="#A020F0">#include &lt;petscdevice.h&gt;</font>
<a name="line17"> 17: </a><font color="#A020F0">#include &lt;petsc/private/petsclegacycupmblas.h&gt;</font>

<a name="line19"> 19: </a><font color="#B22222">/* MANSEC = Sys */</font>

<a name="line21"> 21: </a><font color="#A020F0">#if defined(PETSC_HAVE_CUDA)</font>

<a name="line23"> 23: </a><font color="#B22222">/* complex single */</font>
<a name="line24"> 24: </a><font color="#A020F0">#if defined(PETSC_USE_COMPLEX)</font>
<a name="line25"> 25: </a><font color="#A020F0">#if defined(PETSC_USE_REAL_SINGLE)</font>
<a name="line26"> 26: </a><strong><font color="#228B22">#define cublasXgetrfBatched(a,b,c,d,e,f,g) cublasCgetrfBatched((a),(b),(cuComplex**)(c),(d),(e),(f),(g))</font></strong>
<a name="line27"> 27: </a><strong><font color="#228B22">#define cublasXgetrsBatched(a,b,c,d,e,f,g,h,i,j,k) cublasCgetrsBatched((a),(b),(c),(d),(const cuComplex**)(e),(f),(g),(cuComplex**)(h),(i),(j),(k))</font></strong>
<a name="line28"> 28: </a><font color="#A020F0">#else </font><font color="#B22222">/* complex double */</font><font color="#A020F0"></font>
<a name="line29"> 29: </a><strong><font color="#228B22">#define cublasXgetrfBatched(a,b,c,d,e,f,g) cublasZgetrfBatched((a),(b),(cuDoubleComplex**)(c),(d),(e),(f),(g))</font></strong>
<a name="line30"> 30: </a><strong><font color="#228B22">#define cublasXgetrsBatched(a,b,c,d,e,f,g,h,i,j,k) cublasZgetrsBatched((a),(b),(c),(d),(const cuDoubleComplex**)(e),(f),(g),(cuDoubleComplex**)(h),(i),(j),(k))</font></strong>
<a name="line31"> 31: </a><font color="#A020F0">#endif</font>
<a name="line32"> 32: </a><font color="#A020F0">#else </font><font color="#B22222">/* real single */</font><font color="#A020F0"></font>
<a name="line33"> 33: </a><font color="#A020F0">#if defined(PETSC_USE_REAL_SINGLE)</font>
<a name="line34"> 34: </a><strong><font color="#228B22">#define cublasXgetrfBatched cublasSgetrfBatched</font></strong>
<a name="line35"> 35: </a><strong><font color="#228B22">#define cublasXgetrsBatched cublasSgetrsBatched</font></strong>
<a name="line36"> 36: </a><font color="#A020F0">#else </font><font color="#B22222">/* real double */</font><font color="#A020F0"></font>
<a name="line37"> 37: </a><strong><font color="#228B22">#define cublasXgetrfBatched cublasDgetrfBatched</font></strong>
<a name="line38"> 38: </a><strong><font color="#228B22">#define cublasXgetrsBatched cublasDgetrsBatched</font></strong>
<a name="line39"> 39: </a><font color="#A020F0">#endif</font>
<a name="line40"> 40: </a><font color="#A020F0">#endif</font>

<a name="line42"> 42: </a><font color="#B22222">/* the following ones are used for <a href="https://petsc.org/release/manualpages/Sys/PetscComplex/">PetscComplex</a> in both real and complex scalars */</font>
<a name="line43"> 43: </a><font color="#A020F0">#if defined(PETSC_USE_REAL_SINGLE)</font>
<a name="line44"> 44: </a><strong><font color="#228B22">#define cublasXCaxpy(a,b,c,d,e,f,g)                cublasCaxpy((a),(b),(const cuComplex *)(c),(const cuComplex *)(d),(e),(cuComplex *)(f),(g))</font></strong>
<a name="line45"> 45: </a><strong><font color="#228B22">#define cublasXCgemm(a,b,c,d,e,f,g,h,i,j,k,l,m,n)  cublasCgemm((a),(b),(c),(d),(e),(f),(const cuComplex *)(g),(const cuComplex *)(h),(i),(const cuComplex *)(j),(k),(const cuComplex *)(l),(cuComplex *)(m),(n))</font></strong>
<a name="line46"> 46: </a><strong><font color="#228B22">#define cublasXCscal(a,b,c,d,e)                    cublasCscal((a),(b),(const cuComplex *)(c),(cuComplex *)(d),(e))</font></strong>
<a name="line47"> 47: </a><font color="#A020F0">#else</font>
<a name="line48"> 48: </a><strong><font color="#228B22">#define cublasXCaxpy(a,b,c,d,e,f,g)                cublasZaxpy((a),(b),(const cuDoubleComplex *)(c),(const cuDoubleComplex *)(d),(e),(cuDoubleComplex *)(f),(g))</font></strong>
<a name="line49"> 49: </a><strong><font color="#228B22">#define cublasXCgemm(a,b,c,d,e,f,g,h,i,j,k,l,m,n)  cublasZgemm((a),(b),(c),(d),(e),(f),(const cuDoubleComplex *)(g),(const cuDoubleComplex *)(h),(i),(const cuDoubleComplex *)(j),(k),(const cuDoubleComplex *)(l),(cuDoubleComplex *)(m),(n))</font></strong>
<a name="line50"> 50: </a><strong><font color="#228B22">#define cublasXCscal(a,b,c,d,e)                    cublasZscal((a),(b),(const cuDoubleComplex *)(c),(cuDoubleComplex *)(d),(e))</font></strong>
<a name="line51"> 51: </a><font color="#A020F0">#endif</font>

<a name="line53"> 53: </a><font color="#A020F0">#endif // PETSC_HAVE_CUDA</font>

<a name="line55"> 55: </a><font color="#A020F0">#if defined(PETSC_HAVE_HIP)</font>

<a name="line57"> 57: </a><font color="#B22222">/* complex single */</font>
<a name="line58"> 58: </a><font color="#A020F0">#if defined(PETSC_USE_COMPLEX)</font>
<a name="line59"> 59: </a><font color="#A020F0">#if defined(PETSC_USE_REAL_SINGLE)</font>
<a name="line60"> 60: </a><font color="#A020F0">#else </font><font color="#B22222">/* complex double */</font><font color="#A020F0"></font>
<a name="line61"> 61: </a><font color="#A020F0">#endif</font>
<a name="line62"> 62: </a><font color="#A020F0">#else </font><font color="#B22222">/* real single */</font><font color="#A020F0"></font>
<a name="line63"> 63: </a><font color="#A020F0">#if defined(PETSC_USE_REAL_SINGLE)</font>
<a name="line64"> 64: </a><font color="#A020F0">#else </font><font color="#B22222">/* real double */</font><font color="#A020F0"></font>
<a name="line65"> 65: </a><font color="#A020F0">#endif</font>
<a name="line66"> 66: </a><font color="#A020F0">#endif</font>

<a name="line68"> 68: </a><font color="#A020F0">#endif // PETSC_HAVE_HIP</font>
</pre>
</body>

</html>