Skip to content

Commit ba8e771

Browse files
author
Dhairya Gandhi
committed
let the user decide
1 parent dc8d734 commit ba8e771

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

src/NNlib.jl

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,14 @@ using Requires
55
include("dim_helpers.jl")
66

77
# NNPACK support
8-
if Sys.islinux() || Sys.isapple()
9-
# include("nnpack/NNPACK.jl")
10-
is_nnpack_available() = false
8+
if get(ENV, "NNLIB_USE_NNPACK", "false")
9+
if Sys.islinux() || Sys.isapple()
10+
include("nnpack/NNPACK.jl")
11+
else
12+
is_nnpack_available() = false
13+
end
1114
else
12-
is_nnpack_available() = false
15+
is_nnpack_available() = false
1316
end
1417

1518
include("activation.jl")

0 commit comments

Comments
 (0)