Commit Graph
3 Commits
Author SHA1 Message Date
EthanfelandClaude Sonnet 4.6 8e955187fa style: fix E702 multiple-statements-on-one-line in kernel.py
Split all semicolon-separated statements onto individual lines
throughout the Triton SVD kernels (_svd2_kernel, _svd3_kernel).

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-04-02 01:01:50 +02:00
marduk191andClaude Opus 4.6 bc8e1a9349 feat: add Triton fallback logging in kernel.py
Log info message when Triton is not available so users know
fused N=2/N=3 kernels are disabled.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-03-26 13:37:34 +01:00
marduk191 feceb5e96e feat: add Triton SVD kernel for batched thin-matrix SVD
Fused Triton kernels for N=2,3 (3850-5488x faster than torch),
Gram-eigh hybrid for N>=4, Newton-Schulz inverse sqrt, and
subspace-preserving Procrustes alignment.

From PR #4.
2026-03-26 12:13:33 +01:00