From 2c74976602d5035f84a0901989173f499be9ad70 Mon Sep 17 00:00:00 2001
From: Mike J Innes <mike.j.innes@gmail.com>
Date: Tue, 27 Feb 2018 01:25:40 +0000
Subject: [PATCH] more general

---
 src/treelike.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/treelike.jl b/src/treelike.jl
index cc7d827a..238eb02c 100644
--- a/src/treelike.jl
+++ b/src/treelike.jl
@@ -45,10 +45,10 @@ params(m...) = params(m)
 
 cpu(x) = adapt(Array, x)
 
-default_adaptor = Array
+default_adaptor = identity
 
 @require CuArrays begin
-  global default_adaptor = CuArray
+  global default_adaptor = CuArrays.cu
 end
 
 gpu(x) = adapt(default_adaptor, x)