Skip to content

Commit b1a594b

Browse files
committed
reorganize algorithm unions
1 parent d616e6b commit b1a594b

2 files changed

Lines changed: 12 additions & 16 deletions

File tree

src/implementations/svd.jl

Lines changed: 0 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -284,20 +284,6 @@ end
284284
# placed here to avoid code duplication since much of the logic is replicable across
285285
# CUDA and AMDGPU
286286
###
287-
const CUSOLVER_SVDAlgorithm = Union{
288-
CUSOLVER_QRIteration,
289-
CUSOLVER_SVDPolar,
290-
CUSOLVER_Jacobi,
291-
CUSOLVER_Randomized,
292-
}
293-
const ROCSOLVER_SVDAlgorithm = Union{
294-
ROCSOLVER_QRIteration,
295-
ROCSOLVER_Jacobi,
296-
}
297-
const GPU_SVDAlgorithm = Union{CUSOLVER_SVDAlgorithm, ROCSOLVER_SVDAlgorithm}
298-
299-
const GPU_SVDPolar = Union{CUSOLVER_SVDPolar}
300-
const GPU_Randomized = Union{CUSOLVER_Randomized}
301287

302288
function check_input(
303289
::typeof(svd_trunc!), A::AbstractMatrix, USVᴴ, alg::CUSOLVER_Randomized

src/interface/decompositions.jl

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -235,6 +235,10 @@ Divide and Conquer algorithm.
235235
"""
236236
@algdef CUSOLVER_DivideAndConquer
237237

238+
const CUSOLVER_SVDAlgorithm = Union{
239+
CUSOLVER_QRIteration, CUSOLVER_SVDPolar, CUSOLVER_Jacobi, CUSOLVER_Randomized,
240+
}
241+
238242
# =========================
239243
# ROCSOLVER ALGORITHMS
240244
# =========================
@@ -282,15 +286,21 @@ Divide and Conquer algorithm.
282286
"""
283287
@algdef ROCSOLVER_DivideAndConquer
284288

289+
const ROCSOLVER_SVDAlgorithm = Union{ROCSOLVER_QRIteration, ROCSOLVER_Jacobi}
285290

286291
const GPU_Simple = Union{CUSOLVER_Simple}
287292
const GPU_EigAlgorithm = Union{GPU_Simple}
288293
const GPU_QRIteration = Union{CUSOLVER_QRIteration, ROCSOLVER_QRIteration}
289294
const GPU_Jacobi = Union{CUSOLVER_Jacobi, ROCSOLVER_Jacobi}
290295
const GPU_DivideAndConquer = Union{CUSOLVER_DivideAndConquer, ROCSOLVER_DivideAndConquer}
291296
const GPU_Bisection = Union{ROCSOLVER_Bisection}
292-
const GPU_SVDAlgorithm = Union{GPU_Jacobi, GPU_DivideAndConquer, GPU_QRIteration}
293-
const GPU_EighAlgorithm = Union{GPU_QRIteration, GPU_Jacobi, GPU_DivideAndConquer, GPU_Bisection, CUSOLVER_SVDPolar}
297+
const GPU_EighAlgorithm = Union{
298+
GPU_QRIteration, GPU_Jacobi, GPU_DivideAndConquer, GPU_Bisection,
299+
}
300+
const GPU_SVDAlgorithm = Union{CUSOLVER_SVDAlgorithm, ROCSOLVER_SVDAlgorithm}
301+
302+
const GPU_SVDPolar = Union{CUSOLVER_SVDPolar}
303+
const GPU_Randomized = Union{CUSOLVER_Randomized}
294304

295305
left_orth_kind(::GPU_SVDAlgorithm) = left_orth_svd!
296306
right_orth_kind(::GPU_SVDAlgorithm) = right_orth_svd!

0 commit comments

Comments
 (0)