Crate t4a_cubecl_zspace 

Common ZSpace Utilities for CubeCL

This is a new/experimental module.

The goal will be to unify:

• shape/stride construction/validation.
• stride map view transformations.
• common Shape types.
• common Shape/Size/Reshape utility traits.

The intention is to publish this as a stand-alone zspace module, with no direct tie to cubecl; once it is more polished.

Re-exports

pub use crate::errors::ExpressionError;

Modules

errors

Common Burn Errors

indexing

Indexing Utilities

metadata

striding

Stride Utilities

Macros

shape

smallvec

Reexport for use in macros Creates a SmallVec containing the arguments.

strides

Structs

Shape

Reexport to avoid annoying rust-analyzer bug where it imports the module instead of the macro Shape of a tensor.

SmallVec

Reexport for use in macros A Vec-like container that can store a small number of elements inline.

Strides

Tiling

How many fragments each logical dim is stored as. See the module doc.

Enums

MetadataError

Reexport to avoid annoying rust-analyzer bug where it imports the module instead of the macro Error that can occur when attempting to modify shapes.

Constants

MAX_FRAGMENTS

The most fragments one logical dim is stored as.

MAX_LOGICAL_DIMS

The most logical dims a tiling describes: four bits each in a u32.

Functions

calculate_matmul_output

Reexport to avoid annoying rust-analyzer bug where it imports the module instead of the macro Compute the output shape for matrix multiplication with broadcasting support.