burn_fusion/
lib.rs

1#![warn(missing_docs)]
2#![cfg_attr(docsrs, feature(doc_cfg))]
3
4//! # Burn Fusion
5//!
6//! This library is a part of the Burn project. It is a standalone crate that
7//! can be used to perform automatic operation fusion on backends that support it.
8
9#[macro_use]
10extern crate derive_new;
11
12/// Client module exposing types to communicate with the fusion server.
13pub mod client;
14/// Stream module exposing all tensor operations that can be optimized.
15pub mod stream;
16
17/// Search module for stream optimizations.
18pub(crate) mod search;
19
20mod backend;
21mod ops;
22mod server;
23mod tensor;
24
25pub(crate) use server::*;
26
27pub use backend::*;
28pub use ops::NoOp;
29pub use tensor::*;