A few more updates for GPU compatibility for TensorKit

Author: kshyatt

ext/MatrixAlgebraKitAMDGPUExt/MatrixAlgebraKitAMDGPUExt.jl

@@ -127,6 +127,8 @@ function MatrixAlgebraKit._project_hermitian_diag!(A::StridedROCMatrix, B::Strid
     return nothing
 end
 
+MatrixAlgebraKit.project_hermitian_native!(A::Diagonal{T, <:StridedROCVector}, B::Diagonal{T, <:StridedROCVector}, ::Val{anti}) where {T, anti} = MatrixAlgebraKit.project_hermitian_native!(ROCMatrix(A), ROCMatrix(B), Val(anti))
+
 MatrixAlgebraKit.ishermitian_exact(A::StridedROCMatrix) = all(A .== adjoint(A))
 MatrixAlgebraKit.ishermitian_exact(A::Diagonal{T, <:StridedROCVector{T}}) where {T} =
     all(A.diag .== adjoint(A.diag))

ext/MatrixAlgebraKitCUDAExt/MatrixAlgebraKitCUDAExt.jl

@@ -151,6 +151,8 @@ function MatrixAlgebraKit._project_hermitian_diag!(A::StridedCuMatrix, B::Stride
     return nothing
 end
 
+MatrixAlgebraKit.project_hermitian_native!(A::Diagonal{T, <:StridedCuVector}, B::Diagonal{T, <:StridedCuVector}, ::Val{anti}) where {T, anti} = MatrixAlgebraKit.project_hermitian_native!(CuMatrix(A), CuMatrix(B), Val(anti))
+
 MatrixAlgebraKit.ishermitian_exact(A::StridedCuMatrix) =
     all(A .== adjoint(A))
 MatrixAlgebraKit.ishermitian_exact(A::Diagonal{T, <:StridedCuVector{T}}) where {T} =

src/implementations/projections.jl

@@ -9,13 +9,15 @@ copy_input(::typeof(project_isometric), A) = copy_input(left_polar, A)
 
 function check_input(::typeof(project_hermitian!), A::AbstractMatrix, B::AbstractMatrix, ::AbstractAlgorithm)
     LinearAlgebra.checksquare(A)
-    n = Base.require_one_based_indexing(A)
+    Base.require_one_based_indexing(A)
+    n = size(A, 1)
     B === A || @check_size(B, (n, n))
     return nothing
 end
 function check_input(::typeof(project_antihermitian!), A::AbstractMatrix, B::AbstractMatrix, ::AbstractAlgorithm)
     LinearAlgebra.checksquare(A)
-    n = Base.require_one_based_indexing(A)
+    Base.require_one_based_indexing(A)
+    n = size(A, 1)
     B === A || @check_size(B, (n, n))
     return nothing
 end