Packages

torchx

0.8.0
0.13.0 0.12.0 0.11.0 0.10.2 0.10.1 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.1 0.2.0

LibTorch bindings and backend for Nx

HexDocs

Current section

Files

Jump to
Readme
28 Versions
1 Dependency
Dependants
Files
Activity
torchx lib torchx nif.ex
Raw

lib/torchx/nif.ex 

defmodule Torchx.NIF do

  @moduledoc false

  @on_load :__on_load__



  def __on_load__ do

    case :os.type() do

      {:win32, _} ->

        Torchx.NIF.DLLLoader.add_dll_directory()



      _ ->

        :ok

    end



    path = :filename.join(:code.priv_dir(:torchx), ~c"torchx")

    :erlang.load_nif(path, 0)

  end



  for {op, arity} <- Torchx.__torch__() do

    def unquote(:"#{op}_cpu")(unquote_splicing(Macro.generate_arguments(arity, __MODULE__))),

      do: :erlang.nif_error(:undef)



    def unquote(:"#{op}_io")(unquote_splicing(Macro.generate_arguments(arity, __MODULE__))),

      do: :erlang.nif_error(:undef)

  end



  def mps_is_available(), do: :erlang.nif_error(:undef)

  def cuda_is_available(), do: :erlang.nif_error(:undef)

  def cuda_device_count(), do: :erlang.nif_error(:undef)



  def scalar_type(_tensor), do: :erlang.nif_error(:undef)

  def shape(_tensor), do: :erlang.nif_error(:undef)

  def nbytes(_tensor), do: :erlang.nif_error(:undef)

end