livebook-dev/livebook
1
1
Fork 0
mirror of https://github.com/livebook-dev/livebook.git synced 2025-02-25 23:52:31 +08:00
Code Issues Projects Releases Packages Wiki Activity

Default to Torchx for the Neural Network task when on Windows (#1583)

Browse source
This commit is contained in:
Jonatan Kłosko 2022-12-13 17:12:22 +01:00 committed by GitHub
parent 5fd422c22b
commit 600666fd46
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
lib/livebook/runtime/elixir_standalone.ex
View file

@ -47,6 +47,14 @@ defmodule Livebook.Runtime.ElixirStandalone do
dependency: %{dep: {:exla, "~> 0.4.1"}, config: [nx: [default_backend: EXLA.Backend]]}
}
torchx = %{
name: "torchx",
dependency: %{dep: {:torchx, "~> 0.4.1"}, config: [nx: [default_backend: Torchx.Backend]]}
}
windows? = match?({:win32, _}, :os.type())
nx_backend_package = if(windows?, do: torchx, else: exla)
@extra_smart_cell_definitions [
%{
kind: "Elixir.KinoDB.ConnectionCell",
@ -152,7 +160,7 @@ defmodule Livebook.Runtime.ElixirStandalone do
variants: [
%{
name: "Default",
packages: [kino_bumblebee, exla]
packages: [kino_bumblebee, nx_backend_package]
}
]
}