Skip to main content

ruprim/
lib.rs

1//! Ruda parallel primitives.
2
3#![allow(
4    clippy::needless_range_loop,
5    clippy::too_many_arguments,
6    clippy::type_complexity
7)]
8
9#[cfg(feature = "kernel-ir")]
10pub mod collective;
11
12#[cfg(feature = "kernel-ir")]
13pub mod warp;
14
15#[cfg(feature = "kernel-ir")]
16pub mod block;
17
18#[cfg(feature = "device-primitives")]
19#[allow(unsafe_code)]
20pub mod device;
21
22
23
24#[cfg(feature = "kernel-ir")]
25#[allow(unsafe_code)]
26pub mod reduce;
27
28#[cfg(feature = "elementwise")]
29#[allow(unsafe_code)]
30pub mod elementwise;
31
32#[cfg(feature = "indexing")]
33#[allow(unsafe_code)]
34pub mod indexing;
35
36#[cfg(feature = "tensor-scan")]
37#[allow(unsafe_code)]
38pub mod scan;