fixes for shared library loading and downloading

Author: mortont

lib/ortex/native.ex

@@ -4,6 +4,14 @@ defmodule Ortex.Native do
 
   Stubs for `Rustler` NIFs. These should never be called directly.
   """
+
+  # 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.
+  Rustler.Compiler.compile_crate(__MODULE__, otp_app: :ortex, crate: :ortex)
+  Ortex.Util.copy_ort_libs()
+
   use Rustler,
     otp_app: :ortex,
     crate: :ortex

lib/ortex/util.ex

@@ -0,0 +1,55 @@
+defmodule Ortex.Util do
+  def copy_ort_libs() do
+    rust_env =
+      case Mix.env() do
+        :prod -> "release"
+        _ -> "debug"
+      end
+
+    case :os.type() do
+      {:win32, _} ->
+        Path.wildcard(
+          Path.join([
+            "./_build",
+            Mix.env() |> Atom.to_string(),
+            "lib/ortex/native/ortex",
+            rust_env,
+            "libonnxruntime*.dll*"
+          ])
+        )
+
+      {:unix, :darwin} ->
+        Path.wildcard(
+          Path.join([
+            "./_build/",
+            Mix.env() |> Atom.to_string(),
+            "lib/ortex/native/ortex",
+            rust_env,
+            "libonnxruntime*.dylib*"
+          ])
+        )
+
+      {:unix, _} ->
+        Path.wildcard(
+          Path.join([
+            "./_build/",
+            Mix.env() |> Atom.to_string(),
+            "lib/ortex/native/ortex",
+            rust_env,
+            "libonnxruntime*.so*"
+          ])
+        )
+    end
+    |> Enum.map(fn x ->
+      File.cp!(
+        x,
+        Path.join([
+          "./_build",
+          Mix.env() |> Atom.to_string(),
+          "lib/ortex/priv/native/",
+          Path.basename(x)
+        ])
+      )
+    end)
+  end
+end

mix.exs

@@ -8,10 +8,6 @@ defmodule Ortex.MixProject do
       elixir: "~> 1.14",
       start_permanent: Mix.env() == :prod,
       deps: deps(),
-      compilers: [:copy_ort_libs] ++ Mix.compilers(),
-      aliases: [
-        "compile.copy_ort_libs": &copy_ort_libs/1
-      ],
 
       # Docs
       name: "Ortex",
@@ -54,60 +50,4 @@ defmodule Ortex.MixProject do
       description: "ONNX Runtime bindings for Elixir"
     ]
   end
-
-  defp copy_ort_libs(_) do
-    rust_env =
-      case Mix.env() do
-        :prod -> "release"
-        _ -> "debug"
-      end
-
-    case :os.type() do
-      {:win32, _} ->
-        Path.wildcard(
-          Path.join([
-            "./_build",
-            Mix.env() |> Atom.to_string(),
-            "lib/ortex/native/ortex",
-            rust_env,
-            "libonnxruntime*.dll*"
-          ])
-        )
-
-      {:unix, :darwin} ->
-        Path.wildcard(
-          Path.join([
-            "./_build/",
-            Mix.env() |> Atom.to_string(),
-            "lib/ortex/native/ortex",
-            rust_env,
-            "libonnxruntime*.dylib*"
-          ])
-        )
-
-      {:unix, _} ->
-        Path.wildcard(
-          Path.join([
-            "./_build/",
-            Mix.env() |> Atom.to_string(),
-            "lib/ortex/native/ortex",
-            rust_env,
-            "libonnxruntime*.so*"
-          ])
-        )
-    end
-    |> Enum.map(fn x ->
-      File.cp!(
-        x,
-        Path.join([
-          "./_build",
-          Mix.env() |> Atom.to_string(),
-          "lib/ortex/priv/native/",
-          Path.basename(x)
-        ])
-      )
-    end)
-
-    {:ok, []}
-  end
 end