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 logger.ex
Raw

lib/exla/logger.ex 

defmodule EXLA.Logger do

  @moduledoc false

  use GenServer

  require Logger



  def start_link(_opts) do

    GenServer.start_link(__MODULE__, :ok, name: __MODULE__)

  end



  @impl true

  def init(:ok) do

    :ok = EXLA.NIF.start_log_sink(self())

    {:ok, :unused_state}

  end



  @impl true

  def handle_info({:info, msg, file, line}, state) do

    Logger.info(msg, domain: [:xla], file: file, line: line)

    {:noreply, state}

  end



  @impl true

  def handle_info({:warning, msg, file, line}, state) do

    Logger.warning(msg, domain: [:xla], file: file, line: line)

    {:noreply, state}

  end



  @impl true

  def handle_info({:error, msg, file, line}, state) do

    Logger.error(msg, domain: [:xla], file: file, line: line)

    {:noreply, state}

  end

end