3
3
# # index type
4
4
5
5
function Base. convert (:: Type{cusparseIndexType_t} , T:: DataType )
6
- if T == Int16
7
- return CUSPARSE_INDEX_16U
8
- elseif T == Int32
6
+ if T == Int32
9
7
return CUSPARSE_INDEX_32I
10
8
elseif T == Int64
11
9
return CUSPARSE_INDEX_64I
@@ -15,9 +13,7 @@ function Base.convert(::Type{cusparseIndexType_t}, T::DataType)
15
13
end
16
14
17
15
function Base. convert (:: Type{Type} , T:: cusparseIndexType_t )
18
- if T == CUSPARSE_INDEX_16U
19
- return Int16
20
- elseif T == CUSPARSE_INDEX_32I
16
+ if T == CUSPARSE_INDEX_32I
21
17
return Int32
22
18
elseif T == CUSPARSE_INDEX_64I
23
19
return Int64
@@ -35,7 +31,7 @@ function Base.convert(::Type{cusparseIndexBase_t}, base::Integer)
35
31
elseif base == 1
36
32
return CUSPARSE_INDEX_BASE_ONE
37
33
else
38
- throw (ArgumentError (" CUSPARSE does not support index base $base! " ))
34
+ throw (ArgumentError (" CUSPARSE does not support index base $( base) !" ))
39
35
end
40
36
end
41
37
@@ -45,7 +41,7 @@ function Base.convert(T::Type{<:Integer}, base::cusparseIndexBase_t)
45
41
elseif base == CUSPARSE_INDEX_BASE_ONE
46
42
return T (1 )
47
43
else
48
- throw (ArgumentError (" Unknown index base $base! " ))
44
+ throw (ArgumentError (" Unknown index base $( base) !" ))
49
45
end
50
46
end
51
47
@@ -54,105 +50,105 @@ end
54
50
55
51
function Base. convert (:: Type{cusparseOperation_t} , trans:: SparseChar )
56
52
if trans == ' N'
57
- CUSPARSE_OPERATION_NON_TRANSPOSE
53
+ return CUSPARSE_OPERATION_NON_TRANSPOSE
58
54
elseif trans == ' T'
59
- CUSPARSE_OPERATION_TRANSPOSE
55
+ return CUSPARSE_OPERATION_TRANSPOSE
60
56
elseif trans == ' C'
61
- CUSPARSE_OPERATION_CONJUGATE_TRANSPOSE
57
+ return CUSPARSE_OPERATION_CONJUGATE_TRANSPOSE
62
58
else
63
59
throw (ArgumentError (" Unknown operation $trans " ))
64
60
end
65
61
end
66
62
67
63
function Base. convert (:: Type{cusparseMatrixType_t} , mattype:: SparseChar )
68
64
if mattype == ' G'
69
- CUSPARSE_MATRIX_TYPE_GENERAL
65
+ return CUSPARSE_MATRIX_TYPE_GENERAL
70
66
elseif mattype == ' T'
71
- CUSPARSE_MATRIX_TYPE_TRIANGULAR
67
+ return CUSPARSE_MATRIX_TYPE_TRIANGULAR
72
68
elseif mattype == ' S'
73
- CUSPARSE_MATRIX_TYPE_SYMMETRIC
69
+ return CUSPARSE_MATRIX_TYPE_SYMMETRIC
74
70
elseif mattype == ' H'
75
- CUSPARSE_MATRIX_TYPE_HERMITIAN
71
+ return CUSPARSE_MATRIX_TYPE_HERMITIAN
76
72
else
77
73
throw (ArgumentError (" Unknown matrix type $mattype " ))
78
74
end
79
75
end
80
76
81
77
function Base. convert (:: Type{cusparseSpMatAttribute_t} , attribute:: SparseChar )
82
78
if attribute == ' F'
83
- CUSPARSE_SPMAT_FILL_MODE
79
+ return CUSPARSE_SPMAT_FILL_MODE
84
80
elseif attribute == ' D'
85
- CUSPARSE_SPMAT_DIAG_TYPE
81
+ return CUSPARSE_SPMAT_DIAG_TYPE
86
82
else
87
83
throw (ArgumentError (" Unknown attribute $attribute " ))
88
84
end
89
85
end
90
86
91
87
function Base. convert (:: Type{cusparseFillMode_t} , uplo:: SparseChar )
92
88
if uplo == ' U'
93
- CUSPARSE_FILL_MODE_UPPER
89
+ return CUSPARSE_FILL_MODE_UPPER
94
90
elseif uplo == ' L'
95
- CUSPARSE_FILL_MODE_LOWER
91
+ return CUSPARSE_FILL_MODE_LOWER
96
92
else
97
93
throw (ArgumentError (" Unknown fill mode $uplo " ))
98
94
end
99
95
end
100
96
101
97
function Base. convert (:: Type{cusparseDiagType_t} , diag:: SparseChar )
102
98
if diag == ' U'
103
- CUSPARSE_DIAG_TYPE_UNIT
99
+ return CUSPARSE_DIAG_TYPE_UNIT
104
100
elseif diag == ' N'
105
- CUSPARSE_DIAG_TYPE_NON_UNIT
101
+ return CUSPARSE_DIAG_TYPE_NON_UNIT
106
102
else
107
103
throw (ArgumentError (" Unknown diag type $diag " ))
108
104
end
109
105
end
110
106
111
107
function Base. convert (:: Type{cusparseIndexBase_t} , index:: SparseChar )
112
108
if index == ' Z'
113
- CUSPARSE_INDEX_BASE_ZERO
109
+ return CUSPARSE_INDEX_BASE_ZERO
114
110
elseif index == ' O'
115
- CUSPARSE_INDEX_BASE_ONE
111
+ return CUSPARSE_INDEX_BASE_ONE
116
112
else
117
- throw (ArgumentError (" Unknown index base" ))
113
+ throw (ArgumentError (" Unknown index base $index " ))
118
114
end
119
115
end
120
116
121
117
function Base. convert (:: Type{cusparseDirection_t} , dir:: SparseChar )
122
118
if dir == ' R'
123
- CUSPARSE_DIRECTION_ROW
119
+ return CUSPARSE_DIRECTION_ROW
124
120
elseif dir == ' C'
125
- CUSPARSE_DIRECTION_COLUMN
121
+ return CUSPARSE_DIRECTION_COLUMN
126
122
else
127
123
throw (ArgumentError (" Unknown direction $dir " ))
128
124
end
129
125
end
130
126
131
127
function Base. convert (:: Type{cusparseOrder_t} , order:: SparseChar )
132
128
if order == ' R'
133
- CUSPARSE_ORDER_ROW
129
+ return CUSPARSE_ORDER_ROW
134
130
elseif order == ' C'
135
- CUSPARSE_ORDER_COL
131
+ return CUSPARSE_ORDER_COL
136
132
else
137
133
throw (ArgumentError (" Unknown order $order " ))
138
134
end
139
135
end
140
136
141
137
function Base. convert (:: Type{cusparseSpSVUpdate_t} , update:: SparseChar )
142
138
if update == ' G'
143
- CUSPARSE_SPSV_UPDATE_GENERAL
139
+ return CUSPARSE_SPSV_UPDATE_GENERAL
144
140
elseif update == ' D'
145
- CUSPARSE_SPSV_UPDATE_DIAGONAL
141
+ return CUSPARSE_SPSV_UPDATE_DIAGONAL
146
142
else
147
143
throw (ArgumentError (" Unknown update $update " ))
148
144
end
149
145
end
150
146
151
147
function Base. convert (:: Type{cusparseSpSMUpdate_t} , update:: SparseChar )
152
148
if update == ' G'
153
- CUSPARSE_SPSM_UPDATE_GENERAL
149
+ return CUSPARSE_SPSM_UPDATE_GENERAL
154
150
elseif update == ' D'
155
- CUSPARSE_SPSM_UPDATE_DIAGONAL
151
+ return CUSPARSE_SPSM_UPDATE_DIAGONAL
156
152
else
157
153
throw (ArgumentError (" Unknown update $update " ))
158
154
end
0 commit comments