Tests: - CSA: gather_compressed_kv (top-k) + gather_swa_kv + sparse FMHA - HCA: gather_all_compressed_kv + gather_swa_kv + dense FMHA - Verifies shapes, dtypes, and numerical sanity (no NaN/Inf)
Tests: - CSA: gather_compressed_kv (top-k) + gather_swa_kv + sparse FMHA - HCA: gather_all_compressed_kv + gather_swa_kv + dense FMHA - Verifies shapes, dtypes, and numerical sanity (no NaN/Inf)