Crate torch_build

Utilities to link libtorch FFI interface.

Re-exports

pub use cuda::*;

pub use env::*;

pub use library::*;

Modules

cuda

CUDA related types and functions.

env

Environment variables and constatns.

library

Libtorch installation and capabilities.

Functions

build_cpp

Construct the cc::Build to compile C++ source code.

build_cpp_ext

Construct the cc::Build to compile C++ source code with additional options.

build_cuda

Construct the cc::Build to compile CUDA source code.

build_cuda_ext

Construct the cc::Build to compile CUDA source code with additional options.

check_cxx11_abi

Return true of host system uses C++11 ABI. It is used to set the _GLIBCXX_USE_CXX11_ABI macro.

find_or_download_libtorch_dir

Locate the libtorch directory, or try to download libtorch if it does not exist.

libtorch_url

Generates the libtorch download URL according to host operating system.

probe_libtorch

Probe the installation directory of libtorch and its capabilities.