Fix conversion of tuples to GPU

dfdx · Feb 2, 2020 · 268c75d · 268c75d
1 parent a069933
commit 268c75d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/cuda.jl b/src/cuda.jl
@@ -34,6 +34,8 @@ function to_device(device::GPU, x)
         return x
     elseif isa(x, Real)
         return Float32(x)
+    elseif isa(x, Tuple)
+        return ((to_device(device, el) for el in x)...,)
     elseif isempty(flds)
         # primitive or array
         return cu(x)