Packages

nx

0.12.1
0.13.0 0.12.1 0.12.0 0.11.0 0.10.0 0.9.2 0.9.1 0.9.0 0.8.0 0.7.3 0.7.2 0.7.1 0.7.0 0.6.4 0.6.3 0.6.2 0.6.1 0.6.0 0.5.3 0.5.2 0.5.1 0.5.0 0.4.2 0.4.1 0.4.0 0.3.0 0.2.1 0.2.0 0.1.0

Multi-dimensional arrays (tensors) and numerical definitions for Elixir

HexDocs

Current section

Files

Jump to
Readme
29 Versions
2 Dependencies
Dependants
Files
Activity
nx lib nx application.ex
Raw

lib/nx/application.ex 

defmodule Nx.Application do

  @moduledoc false

  use Application



  def start(_type, _args) do

    children = [

      %{id: Nx.Serving.PG, start: {:pg, :start_link, [Nx.Serving.PG]}},

      {Nx.HiddenServing, Nx.Serving.PG}

    ]



    Supervisor.start_link(children, strategy: :one_for_all, name: Nx.Supervisor)

  end

end



defmodule Nx.HiddenServing do

  # Module to connect hidden nodes with serving.

  # It relies on sending private pg messages to Nx serving.

  @moduledoc false

  use GenServer



  @doc false

  def start_link(scope) do

    GenServer.start_link(__MODULE__, scope)

  end



  @impl true

  def init(scope) do

    pid = Process.whereis(scope)

    :ok = :net_kernel.monitor_nodes(true, node_type: :hidden)



    for node <- Node.list(:hidden) do

      send({scope, node}, {:discover, pid})

    end



    {:ok, scope}

  end



  @impl true

  def handle_info({:nodeup, node, _}, scope) when node != node() do

    send({scope, node}, {:discover, Process.whereis(scope)})

    {:noreply, scope}

  end



  def handle_info(_msg, scope) do

    {:noreply, scope}

  end

end