31 lines
863 B
Elixir
31 lines
863 B
Elixir
defmodule EXLA.Application do
|
|
@moduledoc false
|
|
|
|
def start(_args, _type) do
|
|
# We need this in order for the compile NIF to start `ptxas` using a TF Subprocess.
|
|
# The subprocess relies on `waitpid` which fails under normal circumstances because
|
|
# ERTS sets SIGCHLD to SIGIGN.
|
|
case :os.type() do
|
|
{:win32, _} -> :ok
|
|
_ -> :os.set_signal(:sigchld, :default)
|
|
end
|
|
|
|
pool_size = System.schedulers_online()
|
|
|
|
children = [
|
|
EXLA.Logger,
|
|
{NimblePool,
|
|
worker: {EXLA.MLIR.ContextPool, %{pool_size: pool_size}},
|
|
pool_size: pool_size,
|
|
name: EXLA.MLIR.ContextPool,
|
|
lazy: true},
|
|
EXLA.Client,
|
|
EXLA.Defn.Lock,
|
|
EXLA.Defn.LockedCache,
|
|
{Task.Supervisor, name: EXLA.Defn.TaskSupervisor}
|
|
]
|
|
|
|
Supervisor.start_link(children, name: __MODULE__, strategy: :one_for_one)
|
|
end
|
|
end
|