Crate onednn_src

Crate onednn_src 

Source
Expand description

This crate provides the source build of oneAPI Deep Neural Network Library (oneDNN) for Rust projects.

§Usage

Add this crate to your Cargo.toml:

[dependencies]
onednn-src = "0.1.2"

Then, add the extern declaration to your main.rs or lib.rs to ensure the libraries are properly linked:

extern crate onednn_src;

This crate also exports the DEP_DNNL_ROOT environment variable, which points to the root directory of the compiled oneDNN library. You can use this in your build.rs to locate the library.

§Features

  • graph: Enables the graph component.
  • training: Enables training functionality.
  • inference: Enables inference functionality.
  • experimental: Enables experimental features
  • verbose: Enables verbose mode

By default, the training feature is enabled. The training and inference features are mutually exclusive. If neither is specified, training will be used.