Qianfeng Zhang 1304e807fb Update and fix for leeked changes and make the scripts be able to test/benchmark kStoreLSE cases

2026-06-05 15:51:39 +00:00

..

bench_batched_causal.sh

Update to benchmark scripts to consider for using softmax

2025-10-23 10:09:37 +00:00

bench_cross_attention_with_sparsity.sh

Update to the cross_attention test/bench scripts

2026-05-22 14:44:14 +00:00

bench_jagged_causal_local.sh

Update to benchmark scripts to consider for using softmax

2025-10-23 10:09:37 +00:00

bench_jagged_causal_mattn0_full0.sh

Update and fix for leeked changes and make the scripts be able to test/benchmark kStoreLSE cases

2026-06-05 15:51:39 +00:00

bench_jagged_causal_mattn256_full256_sparsity_90.sh

Add scripts for benchmark sparsity 0.9 cases with mattn256 & full256

2026-01-30 10:02:31 +00:00

bench_jagged_causal_mattn256_full256.sh

Update and fix for leeked changes and make the scripts be able to test/benchmark kStoreLSE cases

2026-06-05 15:51:39 +00:00

bench_jagged_causal.sh

Update to benchmark scripts to consider for using softmax

2025-10-23 10:09:37 +00:00

benchmark_hstu_attention.sh

Update to benchmark scripts to consider for using softmax

2025-10-23 10:09:37 +00:00

test_ck_hstu_mask.sh

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

2025-06-22 16:26:53 +00:00

test_cross_attention_with_sparsity.sh

Update and fix for leeked changes and make the scripts be able to test/benchmark kStoreLSE cases

2026-06-05 15:51:39 +00:00

test_group_hstu_attention.sh

Update to support grouped mode hstu attention

2026-03-09 16:15:58 +00:00

test_group_hstu_softmax_attention.sh

Update and fix for leeked changes and make the scripts be able to test/benchmark kStoreLSE cases

2026-06-05 15:51:39 +00:00

test_hstu_attention_hdim96_hdim64.sh

Rename and add scripts for testing hdim96

2025-12-12 16:16:43 +00:00

test_hstu_attention.sh

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

2025-08-12 03:06:35 +00:00

test_hstu_cross_attention.sh

Update and fix for leeked changes and make the scripts be able to test/benchmark kStoreLSE cases

2026-06-05 15:51:39 +00:00

test_hstu_softmax_attention_hdim96_hdim64.sh

Update and fix for leeked changes and make the scripts be able to test/benchmark kStoreLSE cases

2026-06-05 15:51:39 +00:00

test_hstu_softmax_attention.sh

Update and fix for leeked changes and make the scripts be able to test/benchmark kStoreLSE cases

2026-06-05 15:51:39 +00:00

test_jagged_causal_mattn0_full0.sh

Update and fix for leeked changes and make the scripts be able to test/benchmark kStoreLSE cases

2026-06-05 15:51:39 +00:00

test_jagged_causal_mattn256_full0.sh

Update and fix for leeked changes and make the scripts be able to test/benchmark kStoreLSE cases

2026-06-05 15:51:39 +00:00

test_jagged_causal_mattn256_full256.sh

Update and fix for leeked changes and make the scripts be able to test/benchmark kStoreLSE cases

2026-06-05 15:51:39 +00:00