Skip to content

Commit 872c7b9

Browse files
committed
allocating path for GPU
1 parent 9e776a0 commit 872c7b9

2 files changed

Lines changed: 18 additions & 0 deletions

File tree

ext/MatrixAlgebraKitAMDGPUExt/MatrixAlgebraKitAMDGPUExt.jl

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,4 +52,13 @@ function MatrixAlgebraKit.findtruncated_svd(values::StridedROCVector, strategy::
5252
return MatrixAlgebraKit.findtruncated(values, strategy)
5353
end
5454

55+
MatrixAlgebraKit.ishermitian_exact(A::StridedROCMatrix) =
56+
@invoke MatrixAlgebraKit.ishermitian_exact(A::Any)
57+
MatrixAlgebraKit.ishermitian_approx(A::StridedROCMatrix; kwargs...) =
58+
@invoke MatrixAlgebraKit.ishermitian_approx(A::Any; kwargs...)
59+
MatrixAlgebraKit.isantihermitian_exact(A::StridedROCMatrix) =
60+
@invoke MatrixAlgebraKit.isantihermitian_exact(A::Any)
61+
MatrixAlgebraKit.isantihermitian_approx(A::StridedROCMatrix; kwargs...) =
62+
@invoke MatrixAlgebraKit.isantihermitian_approx(A::Any; kwargs...)
63+
5564
end

ext/MatrixAlgebraKitCUDAExt/MatrixAlgebraKitCUDAExt.jl

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,4 +58,13 @@ function MatrixAlgebraKit.findtruncated_svd(values::StridedCuVector, strategy::T
5858
return MatrixAlgebraKit.findtruncated(values, strategy)
5959
end
6060

61+
MatrixAlgebraKit.ishermitian_exact(A::StridedCuMatrix) =
62+
@invoke MatrixAlgebraKit.ishermitian_exact(A::Any)
63+
MatrixAlgebraKit.ishermitian_approx(A::StridedCuMatrix; kwargs...) =
64+
@invoke MatrixAlgebraKit.ishermitian_approx(A::Any; kwargs...)
65+
MatrixAlgebraKit.isantihermitian_exact(A::StridedCuMatrix) =
66+
@invoke MatrixAlgebraKit.isantihermitian_exact(A::Any)
67+
MatrixAlgebraKit.isantihermitian_approx(A::StridedCuMatrix; kwargs...) =
68+
@invoke MatrixAlgebraKit.isantihermitian_approx(A::Any; kwargs...)
69+
6170
end

0 commit comments

Comments
 (0)