Skip to main content

burn_fusion/stream/
mod.rs

1pub(crate) mod execution;
2pub(crate) mod queue;
3pub(crate) mod shared_tensors;
4pub(crate) mod store;
5
6#[cfg(feature = "memory-checks")]
7/// Memory checks module.
8pub mod memory_checks;
9
10#[cfg(not(feature = "memory-checks"))]
11#[macro_export]
12/// Export memory checks tests.
13macro_rules! memory_checks {
14    () => {
15        #[cfg(test)]
16        mod memory_checks {
17            #[ignore = "'memory-checks' disabled"]
18            #[test]
19            fn test_memory_leaks() {
20                //
21            }
22        }
23    };
24}
25
26mod base;
27mod context;
28mod multi;
29
30pub use base::*;
31pub use context::*;
32pub use execution::*;
33pub use multi::*;