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

Check CuArrays major version.

Browse Source
This commit is contained in:
Tim Besard 2019-03-22 14:12:04 +01:00
parent bc06861320
commit 0734eeb50e
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/cuda/cuda.jl
View File

@ -5,14 +5,14 @@ import ..CuArrays.CUDAdrv: CuPtr, CU_NULL
using Pkg.TOML using Pkg.TOML
function version_check() function version_check()
minor_version = 9 major_version = 1
project = joinpath(dirname(pathof(CuArrays)), "../Project.toml") project = joinpath(dirname(pathof(CuArrays)), "../Project.toml")
project = TOML.parse(String(read(project))) project = TOML.parse(String(read(project)))
version = VersionNumber(get(project, "version", "0.0.0")) version = VersionNumber(get(project, "version", "0.0.0"))
if !(version.major == 0 && version.minor == minor_version) if version.major != major_version
@warn """ @warn """
Flux is only supported with CuArrays v0.$minor_version. Flux is only supported with CuArrays v$major_version.x.
Try running `] pin CuArrays@0.$minor_version`. Try running `] pin CuArrays@$major_version`.
""" """
end end
end end