Packages

exla

0.4.0
0.13.0 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.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.3 0.2.2 0.2.1 0.2.0

Google's XLA (Accelerated Linear Algebra) compiler/backend for Nx

HexDocs

Current section

Files

Jump to
Readme
28 Versions
3 Dependencies
Dependants
Files
Activity
exla lib exla defn runner.ex
Raw

lib/exla/defn/runner.ex 

defmodule EXLA.Defn.Runner do

  # Executes a given function and then exits once the value is read once.

  # This is similar to a task but we don't use a task to avoid polluting

  # inboxes

  @moduledoc false

  use GenServer



  def start_link(ref, fun) do

    GenServer.start_link(__MODULE__, {ref, fun})

  end



  def read(pid) do

    GenServer.call(pid, :read, :infinity)

  end



  @impl true

  def init({ref, fun}) do

    {:ok, nil, {:continue, {ref, fun}}}

  end



  @impl true

  def handle_continue({ref, fun}, nil) do

    receive do

      ^ref -> {:noreply, fun.()}

    end

  end



  @impl true

  def handle_call(:read, _from, value), do: {:stop, :normal, value, value}

end