Merge pull request #142 from yuehhua/test

Split cuda tests and fix GraphSignals.adjacency_matrix
JuliaOrphanage · Sep 15, 2023 · 48fab4f · 48fab4f
2 parents dbf640e + e332b4d
commit 48fab4f
Show file tree

Hide file tree

Showing 8 changed files with 367 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