more general

2018-02-27 01:25:40 +00:00 · 2018-02-27 01:25:40 +00:00 · 2c74976602
commit 2c74976602
parent 466b5c501a
1 changed files with 2 additions and 2 deletions
--- 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)