Skip to content

Commit 3ea43a3

Browse files
author
Katharine Hyatt
committed
Fix AMD extension
1 parent 15feb46 commit 3ea43a3

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

ext/MatrixAlgebraKitAMDGPUExt/MatrixAlgebraKitAMDGPUExt.jl

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -49,8 +49,6 @@ function MatrixAlgebraKit.default_eigh_algorithm(::Type{Base.ReshapedArray{T,2,S
4949
return ROCSOLVER_DivideAndConquer(; kwargs...)
5050
end
5151

52-
MatrixAlgebraKit.ishermitian_exact(A::StridedROCMatrix) = ishermitian(A)
53-
5452
_gpu_geqrf!(A::StridedROCMatrix) = YArocSOLVER.geqrf!(A)
5553
_gpu_ungqr!(A::StridedROCMatrix, τ::StridedROCVector) = YArocSOLVER.ungqr!(A, τ)
5654
_gpu_unmqr!(side::AbstractChar, trans::AbstractChar, A::StridedROCMatrix, τ::StridedROCVector, C::StridedROCVecOrMat) =
@@ -169,7 +167,7 @@ function MatrixAlgebraKit._avgdiff!(A::StridedROCMatrix, B::StridedROCMatrix)
169167
end
170168
thread_dim = 512
171169
block_dim = cld(length(A), thread_dim)
172-
@cuda groupsize=thread_dim gridsize=block_dim _avgdiff_kernel(A, B)
170+
@roc groupsize=thread_dim gridsize=block_dim _avgdiff_kernel(A, B)
173171
return A, B
174172
end
175173
end

0 commit comments

Comments
 (0)