cubecl-utils-rs 0.1.0

Shared CubeCL helpers: GPU tensors, device-limit queries and validated dispatch geometry.
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
//! User-facing re-exports.
//!
//! `use cubecl_utils_rs::prelude::*;` brings in everything needed to size a
//! dispatch and allocate a tensor.

pub use crate::errors::CubeclUtilsErrors;
pub use crate::layout::{pad_vectors, padded_dim, LINE_SIZE};
pub use crate::limits::{
    checked_cube_count, fits_binding, fits_shared_memory, grid_2d, grid_2d_limited,
    plane_partitions, plane_uniform, resident_workgroups, resolve_workgroup_size, GpuLimits,
};
pub use crate::tensor::GpuTensor;
pub use crate::traits::CubeclFloat;