Merge pull request #25 from gregszumel/ort_upgrade

upgrading to ort 2.0.0-rc.0, onnxruntime 1.17

Author: mortont

lib/ortex/util.ex

@@ -1,5 +1,9 @@
 defmodule Ortex.Util do
   @moduledoc false
+  @doc """
+    Copies the libraries downloaded during the ORT build into a path that
+    Elixir can use
+  """
   def copy_ort_libs() do
     build_root = Path.absname(:code.priv_dir(:ortex)) |> Path.dirname()
 
@@ -9,46 +13,38 @@ defmodule Ortex.Util do
         _ -> "debug"
       end
 
-    case :os.type() do
-      {:win32, _} ->
-        Path.wildcard(
-          Path.join([
-            build_root,
-            "native/ortex",
-            rust_env,
-            "libonnxruntime*.dll*"
-          ])
-        )
+    # where the libonnxruntime files are stored
+    rust_path = Path.join([build_root, "native/ortex", rust_env])
 
-      {:unix, :darwin} ->
-        Path.wildcard(
-          Path.join([
-            build_root,
-            "native/ortex",
-            rust_env,
-            "libonnxruntime*.dylib*"
-          ])
-        )
+    onnx_runtime_paths =
+      case :os.type() do
+        {:win32, _} -> Path.join([rust_path, "libonnxruntime*.dll*"])
+        {:unix, :darwin} -> Path.join([rust_path, "libonnxruntime*.dylib*"])
+        {:unix, _} -> Path.join([rust_path, "libonnxruntime*.so*"])
+      end
+      |> Path.wildcard()
 
-      {:unix, _} ->
-        Path.wildcard(
-          Path.join([
-            build_root,
-            "native/ortex",
-            rust_env,
-            "libonnxruntime*.so*"
-          ])
-        )
-    end
+    # where we need to copy the paths
+    destination_dir = Path.join([:code.priv_dir(:ortex), "native"])
+
+    onnx_runtime_paths
     |> Enum.map(fn x ->
-      File.cp!(
-        x,
-        Path.join([
-          :code.priv_dir(:ortex),
-          "native",
-          Path.basename(x)
-        ])
-      )
+      File.cp!(x, Path.join([destination_dir, Path.basename(x)]))
     end)
+
+    # Currently ORT doesn't write the .so file we need (fix incoming https://github.com/pykeio/ort/commit/634e49ab7c960782cc2fb83d84cc219e7bb4ae1f),
+    # so we're hacking a fix here
+    onnx_runtime_filenames = Enum.map(onnx_runtime_paths, &Path.basename/1)
+
+    case "libonnxruntime.so.1.17.0" in onnx_runtime_filenames do
+      true ->
+        nil
+
+      false ->
+        File.cp!(
+          Path.join([destination_dir, "libonnxruntime.so"]),
+          Path.join([destination_dir, "libonnxruntime.so.1.17.0"])
+        )
+    end
   end
 end