cubecl_zspace/lib.rs
1//! # Common `ZSpace` Utilities for `CubeCL`
2//!
3//! This is a new/experimental module.
4//!
5//! The goal will be to unify:
6//! - shape/stride construction/validation.
7//! - stride map view transformations.
8//! - common Shape types.
9//! - common Shape/Size/Reshape utility traits.
10//!
11//! The intention is to publish this as a stand-alone `zspace` module,
12//! with no direct tie to `cubecl`; once it is more polished.
13
14#![no_std]
15
16extern crate alloc;
17
18pub mod errors;
19pub mod indexing;
20pub mod striding;
21
22pub(crate) const INLINE_DIMS: usize = 5;
23
24pub mod metadata;
25mod shape;
26mod strides;
27
28/// Reexport to avoid annoying rust-analyzer bug where it imports the module instead of the macro
29pub use shape::*;
30pub use strides::*;
31
32/// Reexport for use in macros
33pub use smallvec::{SmallVec, smallvec};