update

Author: GiggleLiu

Project.toml

@@ -23,7 +23,7 @@ TupleTools = "9d95972d-f1c8-5527-a6e0-b4b365fa01f6"
 Viznet = "52a3aca4-6234-47fd-b74a-806bdf78ede9"
 
 [compat]
-CUDA = "3.3"
+CUDA = "3.5"
 Compose = "0.9"
 FFTW = "1.4"
 LightGraphs = "1.3"

src/graph_polynomials.jl

@@ -98,11 +98,12 @@ contractx(gp::GraphProblem, x; usecuda=false) = contractf(_->x, gp; usecuda=usec
 function contractf(f, gp::GraphProblem; usecuda=false)
     @debug "generating tensors ..."
     xs = generate_tensors(f, gp)
+    @debug "contracting tensors ..."
     if usecuda
-        xs = CuArray.(xs)
+        gp.code([CuArray(x) for x in xs]...)
+    else
+        gp.code(xs...)
     end
-    @debug "contracting tensors ..."
-    gp.code(xs...)
 end
 
 ############### Problem specific implementations ################