Merge pull request #33 from ranjanan/int32

Support int32

Author: ranjanan

src/aggregate.jl

@@ -1,11 +1,11 @@
 struct StandardAggregation
 end
 
-function aggregation(::StandardAggregation, S)
+function aggregation(::StandardAggregation, S::SparseMatrixCSC{T,R}) where {T,R}
 
     n = size(S, 1)
-    x = zeros(Int, n)
-    y = zeros(Int, n)
+    x = zeros(R, n)
+    y = zeros(R, n)
 
     next_aggregate = 1
 
@@ -96,12 +96,12 @@ function aggregation(::StandardAggregation, S)
     # Some nodes not aggregated
     if minimum(x) == -1
         mask = x .!= -1
-        I = collect(1:n)[mask]
+        I = collect(R, 1:n)[mask]
         J = x[mask] + 1
         V = ones(eltype(S), length(J))
         AggOp = sparse(J,I,V,N,M)
     else
-        Tp = collect(1:n+1)
+        Tp = collect(R, 1:n+1)
         x .= x .+ 1
         Tx = ones(eltype(S), length(x))
         AggOp = SparseMatrixCSC(N, M, Tp, x, Tx)

test/runtests.jl

@@ -266,6 +266,11 @@ end
     test_jacobi_prolongator()
 end
 
+@testset "Int32 support" begin
+    a = sparse(Int32.(1:10), Int32.(1:10), rand(10))
+    @inferred smoothed_aggregation(a)
+end
+
 # Issue #24
 nodes_not_agg()