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

requires update

Browse Source
This commit is contained in:
Mike J Innes 2018-08-03 12:54:24 +01:00
parent a49e2eae41
commit b18b51656c
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/Flux.jl
View File

@ -37,6 +37,6 @@ include("layers/normalise.jl")
include("data/Data.jl") include("data/Data.jl")
@require CuArrays include("cuda/cuda.jl") @init @require CuArrays="3a865a2d-5b23-5a0f-bc46-62713ec82fae" include("cuda/cuda.jl")
end # module end # module

2
src/onehot.jl
View File

@ -32,7 +32,7 @@ import Adapt.adapt
adapt(T, xs::OneHotMatrix) = OneHotMatrix(xs.height, adapt(T, xs.data)) adapt(T, xs::OneHotMatrix) = OneHotMatrix(xs.height, adapt(T, xs.data))
@require CuArrays begin @init @require CuArrays="3a865a2d-5b23-5a0f-bc46-62713ec82fae" begin
import CuArrays: CuArray, cudaconvert import CuArrays: CuArray, cudaconvert
Base.Broadcast._containertype(::Type{<:OneHotMatrix{<:CuArray}}) = CuArray Base.Broadcast._containertype(::Type{<:OneHotMatrix{<:CuArray}}) = CuArray
cudaconvert(x::OneHotMatrix{<:CuArray}) = OneHotMatrix(x.height, cudaconvert(x.data)) cudaconvert(x::OneHotMatrix{<:CuArray}) = OneHotMatrix(x.height, cudaconvert(x.data))

2
src/treelike.jl
View File

@ -53,7 +53,7 @@ cpu(m) = mapleaves(x -> adapt(Array, x), m)
gpu_adaptor = identity gpu_adaptor = identity
@require CuArrays begin @init @require CuArrays="3a865a2d-5b23-5a0f-bc46-62713ec82fae" begin
global gpu_adaptor = CuArrays.cu global gpu_adaptor = CuArrays.cu
end end