2018-01-24 18:45:24 +00:00
|
|
|
module CUDA
|
|
|
|
|
|
2018-08-20 12:08:04 +00:00
|
|
|
using ..CuArrays
|
2018-01-24 18:45:24 +00:00
|
|
|
|
2018-11-06 12:39:54 +00:00
|
|
|
if CuArrays.libcudnn != nothing
|
|
|
|
|
include("cudnn.jl")
|
2018-11-01 04:07:16 +00:00
|
|
|
else
|
2018-11-06 12:39:54 +00:00
|
|
|
@warn("CUDNN is not installed, some functionality will not be available.")
|
2018-11-01 04:07:16 +00:00
|
|
|
end
|
2018-01-24 18:45:24 +00:00
|
|
|
|
|
|
|
|
end
|
