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

cudnn bug is fixed

This commit is contained in:
Mike Innes 2019-09-27 14:15:58 +01:00
parent 46bc8e5e64
commit 691a29cf32
2 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Manifest.toml
View File

@ -106,7 +106,7 @@ version = "4.0.0"
[[CuArrays]]
deps = ["AbstractFFTs", "Adapt", "CEnum", "CUDAapi", "CUDAdrv", "CUDAnative", "DataStructures", "GPUArrays", "LinearAlgebra", "MacroTools", "NNlib", "Printf", "Random", "Requires", "SparseArrays", "TimerOutputs"]
git-tree-sha1 = "4e638627673078c58b6e6bb789937822d83350ff"
git-tree-sha1 = "cc22ec1abd471b4529883a8174944b513d75ab33"
repo-rev = "tb/flux"
repo-url = "https://github.com/JuliaGPU/CuArrays.jl.git"
uuid = "3a865a2d-5b23-5a0f-bc46-62713ec82fae"

8
test/cuda/cuda.jl
View File

@ -51,9 +51,7 @@ end
end
if CuArrays.libcudnn != nothing
@info "Testing Flux/CUDNN"
include("cudnn.jl")
if !haskey(ENV, "CI_DISABLE_CURNN_TEST")
include("curnn.jl")
end
@info "Testing Flux/CUDNN"
include("cudnn.jl")
include("curnn.jl")
end