TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

more general

Browse Source
This commit is contained in:
Mike J Innes 2018-02-27 01:25:40 +00:00
parent 466b5c501a
commit 2c74976602
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/treelike.jl
View File

@ -45,10 +45,10 @@ params(m...) = params(m)
cpu(x) = adapt(Array, x) cpu(x) = adapt(Array, x)
default_adaptor = Array default_adaptor = identity
@require CuArrays begin @require CuArrays begin
global default_adaptor = CuArray global default_adaptor = CuArrays.cu
end end
gpu(x) = adapt(default_adaptor, x) gpu(x) = adapt(default_adaptor, x)