Remove torch version env var

elixir-nx · Nov 4, 2023 · e572aeb · e572aeb
1 parent 849d5d5
commit e572aeb
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -23,7 +23,6 @@ jobs:
     env:
       MIX_ENV: test
       XLA_FLAGS: --xla_force_host_platform_device_count=2
-      LIBTORCH_VERSION: ${{ matrix.torch_version }}
       LIBTORCH_DIR: ${{ github.workspace }}/torchx/cache/libtorch-${{ matrix.torch_version }}
     steps:
       - uses: actions/checkout@v2
@@ -75,7 +74,6 @@ jobs:
     env:
       MIX_ENV: test
       XLA_FLAGS: --xla_force_host_platform_device_count=2
-      LIBTORCH_VERSION: ${{ matrix.torch_version }}
       LIBTORCH_DIR: ${{ github.workspace }}\torchx\cache\libtorch-${{ matrix.torch_version }}
     steps:
       - name: Configure Git

diff --git a/torchx/mix.exs b/torchx/mix.exs
@@ -101,8 +101,7 @@ defmodule Torchx.MixProject do
 
   defp libtorch_config() do
     target = System.get_env("LIBTORCH_TARGET", "cpu")
-    default_version = "2.1.0"
-    version = System.get_env("LIBTORCH_VERSION", default_version)
+    version = System.get_env("LIBTORCH_VERSION", "2.1.0")
     env_dir = System.get_env("LIBTORCH_DIR")
 
     %{