mirror of
https://github.com/livebook-dev/livebook.git
synced 2024-09-20 10:05:57 +08:00
Do not set XLA_TARGET when using Livebook CUDA-based images (#2755)
This commit is contained in:
parent
1ac585c9e1
commit
fbb387e882
|
@ -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
|
||||
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in a new issue