CubeK Reduce

Implements a wide variety of reduction algorithms across multiple instruction sets and hardware targets for efficient tensor reduction.

Running Tests

Important Environment Variables

Test behavior is controlled by the shared CUBE_TEST_MODE env var (see cubek-test-utils).

• CUBE_TEST_MODE=Correct (default): numerical errors fail the test; compilation / hardware-incompatibility errors are accepted.
• CUBE_TEST_MODE=Strict: both numerical and compilation errors fail the test. Useful to surface tests that are silently skipped on your hardware.
• CUBE_TEST_MODE=PrintAll[:<filter>] / PrintFail[:<filter>]: print tensor elements; see cubek-test-utils docs.

Important Feature Flags

• extended: enables the Cube reduction-routine strategy tests. These are slow on CPU and can stall CI, so they're opt-in.
• full: alias for extended (room for future growth).

Examples

# Default (fast) test run on CUDA
cargo test --features cubecl/cuda

# Run the full suite, including Cube-strategy tests
cargo test --features cubecl/cuda,extended

# Fail on any silently-skipped tests
CUBE_TEST_MODE=Strict cargo test --features cubecl/cuda,extended