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

add LinearAlgebra

This commit is contained in:
Yao Lu 2020-04-20 19:54:44 +08:00
parent ba0fca5a19
commit 68b84bba36
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Project.toml
View File

@ -10,6 +10,7 @@ Colors = "5ae59095-9a9b-59fe-a467-6f913c188581"
CuArrays = "3a865a2d-5b23-5a0f-bc46-62713ec82fae" CuArrays = "3a865a2d-5b23-5a0f-bc46-62713ec82fae"
DelimitedFiles = "8bb1440f-4735-579b-a4ab-409b98df4dab" DelimitedFiles = "8bb1440f-4735-579b-a4ab-409b98df4dab"
Juno = "e5e0dc1b-0480-54bc-9374-aad01c23163d" Juno = "e5e0dc1b-0480-54bc-9374-aad01c23163d"
LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
MacroTools = "1914dd2f-81c6-5fcd-8719-6d5c9610ff09" MacroTools = "1914dd2f-81c6-5fcd-8719-6d5c9610ff09"
NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd" NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd"
Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f" Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"

3
src/Flux.jl
View File

@ -4,7 +4,7 @@ module Flux
using Base: tail using Base: tail
using Statistics, Random, LinearAlgebra using Statistics, Random, LinearAlgebra
using Zygote, MacroTools, Juno, Reexport, Requires using Zygote, MacroTools, Juno, Reexport
using MacroTools: @forward using MacroTools: @forward
@reexport using NNlib @reexport using NNlib
using Zygote: Params, @adjoint, gradient, pullback, @nograd using Zygote: Params, @adjoint, gradient, pullback, @nograd
@ -51,6 +51,7 @@ function __init__()
@warn "CuArrays.jl found cuda, but did not find libcudnn. Some functionality will not be available." @warn "CuArrays.jl found cuda, but did not find libcudnn. Some functionality will not be available."
end end
end end
@require MPI="da04e1cc-30fd-572f-bb4f-1f8673147195" include("optimise/mpi.jl")
end end
end # module end # module