split cuda tests and fix GraphSignals.adjacency_matrix

JuliaOrphanage · Sep 13, 2023 · b38166f · b38166f
1 parent dbf640e
commit b38166f
Show file tree

Hide file tree

Showing 8 changed files with 366 additions and 353 deletions.
diff --git a/ext/GraphSignalsCUDAExt/linalg.jl b/ext/GraphSignalsCUDAExt/linalg.jl
@@ -8,9 +8,12 @@ function GraphSignals.adjacency_matrix(adj::CuSparseMatrixCSC)
     return CuMatrix(adj)
 end
 
-adjacency_matrix(adj::CuMatrix{T}, ::Type{T}) where {T} = adjacency_matrix(adj)
+function GraphSignals.adjacency_matrix(adj::CuMatrix{T}, ::Type{S}) where {T,S}
+    GraphSignals._dim_check(adj)
+    return CuMatrix{S}(adj)
+end
 
-function adjacency_matrix(adj::CuMatrix)
+function GraphSignals.adjacency_matrix(adj::CuMatrix)
     GraphSignals._dim_check(adj)
     return adj
 end

diff --git a/test/cuda.jl b/test/cuda.jl