diff --git a/lib/BNNS/BNNS.jl b/lib/BNNS/BNNS.jl index 34fbb43..4f87498 100644 --- a/lib/BNNS/BNNS.jl +++ b/lib/BNNS/BNNS.jl @@ -1,4 +1,3 @@ -module BNNS using BFloat16s include("libBNNS.jl") @@ -41,5 +40,3 @@ function randn end function rand! end function randn! end function seed! end - -end diff --git a/src/AppleAccelerate.jl b/src/AppleAccelerate.jl index 21e944e..7bda319 100644 --- a/src/AppleAccelerate.jl +++ b/src/AppleAccelerate.jl @@ -95,7 +95,12 @@ end include("Util.jl") include("Array.jl") include("DSP.jl") +end + +module BNNS +@static if Sys.isapple() include("../lib/BNNS/BNNS.jl") end +end # module BNNS end # module