fix CountingAll for GPU

Author: GiggleLiu

src/interfaces.jl

@@ -263,7 +263,8 @@ function solve(gp::GenericTensorNetwork, property::AbstractProperty; T=Float64,
         return asarray(post_invert_exponent.(res), res)
     elseif property isa CountingAll
         return big_integer_solve(Int32, 100) do T
-            contractx(gp, one(T); usecuda=usecuda)
+            # NOTE: download to CPU after computation for post-processing (CRT)
+            Array(contractx(gp, one(T); usecuda=usecuda))
         end
     elseif property isa PartitionFunction
         return contractx(gp, exp(property.beta); usecuda=usecuda)
@@ -500,4 +501,4 @@ end
 function Base.findmax(problem::AbstractProblem, solver::GTNSolver)
     res = collect(solve(GenericTensorNetwork(problem; optimizer=solver.optimizer), ConfigsMax(; tree_storage=true); usecuda=solver.usecuda, T=solver.T)[].c)
     return map(x -> ProblemReductions.id_to_config(problem, Int.(x) .+ 1), res)
-end
+end