From 68b84bba368ce0783196aef183dce6bf49343690 Mon Sep 17 00:00:00 2001 From: Yao Lu Date: Mon, 20 Apr 2020 19:54:44 +0800 Subject: [PATCH] add LinearAlgebra --- Project.toml | 1 + src/Flux.jl | 3 ++- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/Project.toml b/Project.toml index 1883d974..7ce3909a 100644 --- a/Project.toml +++ b/Project.toml @@ -10,6 +10,7 @@ Colors = "5ae59095-9a9b-59fe-a467-6f913c188581" CuArrays = "3a865a2d-5b23-5a0f-bc46-62713ec82fae" DelimitedFiles = "8bb1440f-4735-579b-a4ab-409b98df4dab" Juno = "e5e0dc1b-0480-54bc-9374-aad01c23163d" +LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e" MacroTools = "1914dd2f-81c6-5fcd-8719-6d5c9610ff09" NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd" Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f" diff --git a/src/Flux.jl b/src/Flux.jl index 0195cc8c..1ec1ad94 100644 --- a/src/Flux.jl +++ b/src/Flux.jl @@ -4,7 +4,7 @@ module Flux using Base: tail using Statistics, Random, LinearAlgebra -using Zygote, MacroTools, Juno, Reexport, Requires +using Zygote, MacroTools, Juno, Reexport using MacroTools: @forward @reexport using NNlib 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." end end + @require MPI="da04e1cc-30fd-572f-bb4f-1f8673147195" include("optimise/mpi.jl") end end # module