3 people f82c190cbb [kernel] Build + allow attn_qat_infer FP4 attention on sm_121a (DGX Spark) (#1598) ...

Co-authored-by: SolitaryThinker <wlsaidhi@gmail.com>
Co-authored-by: Claude Opus 5.5 (1M context) <noreply@anthropic.com>

2026-10-07 04:09:39 -07:00

..

__init__.py

[kernel] Reorg and fix fastvideo-kernel (#962)

2025-12-26 01:50:24 -06:00

support_flex_sta.py

[kernel] Reorg and fix fastvideo-kernel (#962)

2025-12-26 01:50:24 -06:00

test_attn_qat_infer_sparse.py

[feat]: FastH3 on single RTX GPUs (#1919)

2026-10-06 11:37:01 -07:00

test_attn_qat_infer.py

[kernel] Build + allow attn_qat_infer FP4 attention on sm_121a (DGX Spark) (#1598)

2026-10-07 04:09:39 -07:00

test_attn_qat_train_routing.py

[feat] Port NVFP4 QAT/QAD to modular train framework (#1619)

2026-07-18 15:01:09 -07:00

test_attn_qat_train.py

[bugfix] Fix dead NVFP4 QAT plumbing: bench imports, int8 error string, smooth_q crash (#1723)

2026-09-24 17:09:33 -07:00

test_block_sparse_bwd_sm100a.py

[kernel] sm_100a CUDA backward for VSA block-sparse attention, 128-token blocks (#1866)

2026-09-29 16:57:22 -07:00

test_block_sparse_sm100a_clc.py

[bugfix] sm_100a VSA: fence the CLC response read before the slot release; forward PV descriptor lifetime (#1904)

2026-10-02 12:37:34 -07:00

test_block_sparse_sm100a_dispatch.py

[kernel] sm_100a CUDA backward for VSA block-sparse attention, 128-token blocks (#1866)

2026-09-29 16:57:22 -07:00

test_build_arch.py

[kernel] Build + allow attn_qat_infer FP4 attention on sm_121a (DGX Spark) (#1598)

2026-10-07 04:09:39 -07:00

test_device_specs.py

[perf] benchmark_attn_qat_train: device peak-TFLOPS table instead of silent RTX 5090 default (#1724)

2026-09-24 17:08:03 -07:00

test_fused_compress_topk.py

[misc] yapf pass under CI's interpreter (3.12) + pin hook language_version (#1702)

2026-08-08 22:24:17 -07:00

test_nvfp4_benchmark_imports.py

[bugfix] Fix dead NVFP4 QAT plumbing: bench imports, int8 error string, smooth_q crash (#1723)

2026-09-24 17:09:33 -07:00

test_sta.py

[misc] yapf pass under CI's interpreter (3.12) + pin hook language_version (#1702)

2026-08-08 22:24:17 -07:00

test_turbodiffusion.py

[bugfix] Fix dead NVFP4 QAT plumbing: bench imports, int8 error string, smooth_q crash (#1723)

2026-09-24 17:09:33 -07:00

test_vmoba_correctness.py

[misc] yapf pass under CI's interpreter (3.12) + pin hook language_version (#1702)

2026-08-08 22:24:17 -07:00

test_vsa128_backward.py

[perf] Add FA4 CuTe backward support for VSA-256 (#1639)

2026-08-19 11:46:49 -07:00

test_vsa256_backward.py

[perf] Add FA4 CuTe backward support for VSA-256 (#1639)

2026-08-19 11:46:49 -07:00

test_vsa256_forward_cross.py

[misc] yapf pass under CI's interpreter (3.12) + pin hook language_version (#1702)

2026-08-08 22:24:17 -07:00

test_vsa256_forward_vbs.py

[misc] yapf pass under CI's interpreter (3.12) + pin hook language_version (#1702)

2026-08-08 22:24:17 -07:00

test_vsa256_forward.py

[misc] yapf pass under CI's interpreter (3.12) + pin hook language_version (#1702)

2026-08-08 22:24:17 -07:00

test_vsa256_triton.py

[perf] Add FA4 CuTe backward support for VSA-256 (#1639)

2026-08-19 11:46:49 -07:00

test_vsa_combine.py

[perf] avoid full-sequence materialization in VSA coarse/sparse combine (#1813)

2026-10-07 01:54:35 -07:00

test_vsa_forward.py

[misc] yapf pass under CI's interpreter (3.12) + pin hook language_version (#1702)

2026-08-08 22:24:17 -07:00

test_vsa_triton_backward_scale.py

[bugfix] fastvideo-kernel: fix Triton block-sparse backward logit scaling (bf16 K pre-scaling) (#1730)

2026-08-21 04:50:06 -05:00

test_vsa_utils.py

[misc] yapf pass under CI's interpreter (3.12) + pin hook language_version (#1702)

2026-08-08 22:24:17 -07:00

test_vsa_varlen.py

[perf] Add FA4 CuTe backward support for VSA-256 (#1639)

2026-08-19 11:46:49 -07:00

test_vsa.py

[misc] yapf pass under CI's interpreter (3.12) + pin hook language_version (#1702)

2026-08-08 22:24:17 -07:00

utils.py

[misc] yapf pass under CI's interpreter (3.12) + pin hook language_version (#1702)

2026-08-08 22:24:17 -07:00