strided-perm 0.2.0

Cache-efficient tensor permutation / transpose (HPTT-inspired).
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

//! Architecture-specific micro-kernel trait and dispatch.
//!
//! The micro-kernel is the innermost building block: an N×N in-register
//! transpose where N = REGISTER_BITS / 8 / sizeof(T).

mod scalar;

/// Architecture-specific N×N transpose micro-kernel.
///
/// A micro-kernel transposes a MICRO × MICRO tile:
///   `dst[i + j*ldb] = src[i*lda + j]` for i,j in 0..MICRO
///
/// BLOCK = MICRO * 4 defines the macro-kernel tile size.
pub(crate) trait MicroKernel<T: Copy> {
    /// Micro-tile side length.
    /// e.g. 4 for f64 (scalar/AVX2), 8 for f32 (scalar/AVX2).
    const MICRO: usize;

    /// Macro-tile side length = MICRO * 4.
    const BLOCK: usize;

    /// Transpose a full MICRO × MICRO tile.
    ///
    /// # Safety
    /// - `src` must be readable for MICRO elements along stride-1 and MICRO rows of stride `lda`
    /// - `dst` must be writable for MICRO elements along stride-1 and MICRO rows of stride `ldb`
    unsafe fn transpose_micro(src: *const T, lda: isize, dst: *mut T, ldb: isize);
}

/// Marker type for scalar (non-SIMD) micro-kernels.
pub(crate) struct ScalarKernel;