Qianfeng Zhang ea9e4d8e00 Update to use BottomRight-Diagonal masking when seqlen_kv is bigger than seqlen_q

2026-06-23 09:27:59 +00:00

..

bench_batched_causal.sh

Update to benchmark scripts to consider for using softmax

2026-06-23 09:27:59 +00:00

bench_jagged_causal_local.sh

Update to benchmark scripts to consider for using softmax

2026-06-23 09:27:59 +00:00

bench_jagged_causal_mattn0_full0.sh

Update to benchmark scripts to consider for using softmax

2026-06-23 09:27:59 +00:00

bench_jagged_causal_mattn256_full256.sh

Update to benchmark scripts to consider for using softmax

2026-06-23 09:27:59 +00:00

bench_jagged_causal.sh

Update to benchmark scripts to consider for using softmax

2026-06-23 09:27:59 +00:00

benchmark_hstu_attention.sh

Update to benchmark scripts to consider for using softmax

2026-06-23 09:27:59 +00:00

test_ck_hstu_mask.sh

Update to test_ck_hstu_mask.sh and test_pytorch_hstu_mask.py to align their testings

2026-06-23 09:20:58 +00:00

test_hstu_attention_hdim96_hdim64.sh

Rename and add scripts for testing hdim96

2026-06-23 09:27:59 +00:00

test_hstu_attention_seqlen_kv.sh

Update to use BottomRight-Diagonal masking when seqlen_kv is bigger than seqlen_q

2026-06-23 09:27:59 +00:00

test_hstu_attention.sh

Add norm_dist parameter for hstu example to select either normal or uniform distribution to initialize data

2026-06-23 09:27:58 +00:00

test_hstu_softmax_attention_hdim96_hdim64.sh

Rename and add scripts for testing hdim96

2026-06-23 09:27:59 +00:00

test_hstu_softmax_attention.sh

Add support of softmax in hstu attention

2026-06-23 09:27:59 +00:00

test_jagged_causal_mattn0_full0.sh

Add norm_dist parameter for hstu example to select either normal or uniform distribution to initialize data

2026-06-23 09:27:58 +00:00

test_jagged_causal_mattn256_full0.sh

Add norm_dist parameter for hstu example to select either normal or uniform distribution to initialize data

2026-06-23 09:27:58 +00:00

test_jagged_causal_mattn256_full256.sh

Add norm_dist parameter for hstu example to select either normal or uniform distribution to initialize data

2026-06-23 09:27:58 +00:00