Skip to content

Commit 836889d

Browse files
author
Dhairya Gandhi
committed
make NNPACK conditional
1 parent cc06137 commit 836889d

File tree

1 file changed

+12
-4
lines changed

1 file changed

+12
-4
lines changed

deps/build.jl

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -33,10 +33,18 @@ end
3333

3434
# If we have a download, and we are unsatisfied (or the version we're
3535
# trying to install is not itself installed) then load it up!
36-
if unsatisfied || !isinstalled(dl_info...; prefix=prefix)
37-
# Download and install binaries
36+
# Download and install binaries
37+
use_nnpack = get(ENV, "NNLIB_USE_NNPACK", "false") == "true"
38+
os_support = Sys.islinux() || Sys.isapple()
39+
if use_nnpack && os_support
40+
if unsatisfied || !isinstalled(dl_info...; prefix=prefix)
3841
install(dl_info...; prefix=prefix, force=true, verbose=verbose)
42+
# Write out a deps.jl file that will contain mappings for our products
43+
write_deps_file(joinpath(@__DIR__, "deps.jl"), products, verbose=verbose)
44+
end
45+
else
46+
open(joinpath(@__DIR__, "deps.jl"), "w") do io
47+
write(io, "check_deps() = false")
48+
end
3949
end
4050

41-
# Write out a deps.jl file that will contain mappings for our products
42-
write_deps_file(joinpath(@__DIR__, "deps.jl"), products, verbose=verbose)

0 commit comments

Comments
 (0)