fix: replace missing IREE_DIR

elixir-nx · Jul 14, 2024 · 1ed8230 · 1ed8230
1 parent 52956b7
commit 1ed8230
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/.github/workflows/precompiled_nif.yml b/.github/workflows/precompiled_nif.yml
@@ -38,7 +38,6 @@ jobs:
           mix local.hex --force
           mix local.rebar --force
           echo "NX_IREE_SOURCE_DIR=$(elixir -e 'IO.puts(Path.absname(~s(${{env.NX_IREE_SOURCE_DIR}})))')" > $GITHUB_ENV
-          echo $NX_IREE_SOURCE_DIR
 
       - name: Cache mix dependencies
         id: cache-mix-deps

diff --git a/cmake/CMakeLists.txt b/cmake/CMakeLists.txt
@@ -28,7 +28,7 @@ add_library(${_NAME} SHARED ${iree_runtime_sources})
 
 set(__BUILD_DIR "${IREE_RUNTIME_BUILD_DIR}")
 
-add_subdirectory("${IREE_DIR}" ${__BUILD_DIR} EXCLUDE_FROM_ALL)
+add_subdirectory("${NX_IREE_SOURCE_DIR}" ${__BUILD_DIR} EXCLUDE_FROM_ALL)
 
 install(
     TARGETS ${_NAME}

diff --git a/mix.exs b/mix.exs
@@ -24,7 +24,7 @@ defmodule NxIREE.MixProject do
           "CWD_RELATIVE_TO_PRIV_PATH" => cwd_relative_to_priv,
           "MAKE_NUM_JOBS" => n_jobs,
           "IREE_GIT_REV" => nx_iree_config().tag,
-          "IREE_DIR" => nx_iree_config().source_dir
+          "NX_IREE_SOURCE_DIR" => nx_iree_config().source_dir
         }
       end
     ]