Do not set XLA_TARGET when using Livebook CUDA-based images (#2755)

livebook-dev · Aug 19, 2024 · fbb387e · fbb387e
1 parent 1ac585c
commit fbb387e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 9 deletions.
diff --git a/lib/livebook/config.ex b/lib/livebook/config.ex
@@ -65,11 +65,7 @@ defmodule Livebook.Config do
 
     [
       %{tag: version, name: "Livebook", env: []},
-      %{
-        tag: "#{version}-cuda12",
-        name: "Livebook + CUDA 12",
-        env: [{"XLA_TARGET", "cuda120"}]
-      }
+      %{tag: "#{version}-cuda12", name: "Livebook + CUDA 12", env: []}
     ]
   end
 

diff --git a/test/livebook/hubs/dockerfile_test.exs b/test/livebook/hubs/dockerfile_test.exs
@@ -174,8 +174,6 @@ defmodule Livebook.Hubs.DockerfileTest do
 
       assert dockerfile =~ """
              FROM ghcr.io/livebook-dev/livebook:#{@version}-cuda12
-
-             ENV XLA_TARGET "cuda120"
              """
     end
 
@@ -251,10 +249,9 @@ defmodule Livebook.Hubs.DockerfileTest do
       hub = team_hub()
       agent_key = Livebook.Factory.build(:agent_key)
 
-      %{image: image, env: env} = Dockerfile.online_docker_info(config, hub, agent_key)
+      %{image: image, env: _env} = Dockerfile.online_docker_info(config, hub, agent_key)
 
       assert image == "ghcr.io/livebook-dev/livebook:#{@version}-cuda12"
-      assert {"XLA_TARGET", "cuda120"} in env
     end
 
     test "deploying with auto cluster setup" do