nif: quiet skip-compile and libonnxruntime warnings

Author: serpent213

lib/ortex/native.ex

@@ -1,18 +1,30 @@
 defmodule Ortex.Native do
   @moduledoc false
 
-  @rustler_version Application.spec(:rustler, :vsn) |> to_string() |> Version.parse!()
   @skip_compile? (case System.get_env("ORTEX_SKIP_COMPILE") do
                     nil -> false
                     value -> String.downcase(value) in ["1", "true", "yes", "on"]
                   end)
+  @skip_download? (case System.get_env("ORTEX_SKIP_DOWNLOAD") do
+                     nil -> false
+                     value -> String.downcase(value) in ["1", "true", "yes", "on"]
+                   end)
 
   # We have to compile the crate before `use Rustler` compiles the crate since
   # cargo downloads the onnxruntime shared libraries and they are not available
   # to load or copy into Elixir's during the on_load or Elixir compile steps.
   # In the future, this may be configurable in Rustler.
   if not @skip_compile? do
-    if Version.compare(@rustler_version, "0.30.0") in [:gt, :eq] do
+    if @skip_download? do
+      System.put_env("ORT_SKIP_DOWNLOAD", "1")
+    end
+
+    rustler_version =
+      Application.spec(:rustler, :vsn)
+      |> to_string()
+      |> Version.parse!()
+
+    if Version.compare(rustler_version, "0.30.0") in [:gt, :eq] do
       Rustler.Compiler.compile_crate(:ortex, Application.compile_env(:ortex, __MODULE__, []),
         otp_app: :ortex,
         crate: :ortex

lib/ortex/util.ex

@@ -5,6 +5,7 @@ defmodule Ortex.Util do
     Elixir can use
   """
   def copy_ort_libs() do
+    suppress_warning? = suppress_copy_warning?()
     build_root = Path.absname(:code.priv_dir(:ortex)) |> Path.dirname()
     ort_lib_location = System.get_env("ORT_LIB_LOCATION")
     destination_dir = Path.join([:code.priv_dir(:ortex), "native"])
@@ -37,16 +38,11 @@ defmodule Ortex.Util do
         Destination: #{destination_dir}
         """
 
-      onnx_runtime_paths == [] and test_env?() ->
+      onnx_runtime_paths == [] and (test_env?() or suppress_warning?) ->
         :ok
 
       onnx_runtime_paths == [] ->
-        IO.warn("""
-        Unable to locate libonnxruntime binaries.
-        Searched: #{Enum.join(search_patterns, ", ")}
-        Destination: #{destination_dir}
-        Set ORT_LIB_LOCATION or run mix compile to build the NIF.
-        """)
+        :ok
 
       true ->
         Enum.each(onnx_runtime_paths, fn path ->
@@ -72,7 +68,9 @@ defmodule Ortex.Util do
     expanded = Path.expand(path)
 
     if File.dir?(expanded) do
-      [lib_glob(expanded)]
+      [expanded, Path.join(expanded, "lib"), Path.join(expanded, "lib64")]
+      |> Enum.filter(&File.dir?/1)
+      |> Enum.map(&lib_glob/1)
     else
       [expanded]
     end
@@ -113,4 +111,16 @@ defmodule Ortex.Util do
         System.get_env("MIX_ENV") == "test"
     end
   end
+
+  defp suppress_copy_warning?() do
+    truthy_env?("ORTEX_SKIP_COMPILE") or truthy_env?("ORTEX_SKIP_DOWNLOAD") or
+      truthy_env?("ORT_PREFER_DYNAMIC_LINK")
+  end
+
+  defp truthy_env?(name) do
+    case System.get_env(name) do
+      nil -> false
+      value -> String.downcase(value) in ["1", "true", "yes", "on"]
+    end
+  end
 end