trellis_sparse_mrp_mlx/tests
John a44a0d7360 Derive RoPE phases from coordinates for the sparse SLAT flows
ss_flow ships rope_phases precomputed; the SLAT flows do not, because their positions
are the input SparseTensor's own coordinates and vary per input. rope_phases_from_coords
reproduces RotaryPositionEmbedder, including the pad detail: 3*21=63 frequencies fall one
short of head_dim//2=64 and upstream right-pads with 1+0j, so dropping it would shift
every later pair by a slot.

Verified against ss_flow's own shipped tensor to 9.6e-7 - a real ground truth, since the
same embedder produced it.
2026-08-02 12:05:33 +10:00
..
test_dit.py Derive RoPE phases from coordinates for the sparse SLAT flows 2026-08-02 12:05:33 +10:00
test_ops.py Shared TRELLIS-lineage sparse core in MLX 2026-08-02 10:31:07 +10:00
test_sparse.py Add masked upsample and parameterise downsample reduction (16/16 tests) 2026-08-02 10:42:20 +10:00