Qianfeng Zhang d1505786f8 Add support of softmax in hstu attention

2025-10-20 14:26:55 +00:00

..

bench_batched_causal.sh

Move all test and bench scripts to folder scripts

2025-06-06 08:22:38 +00:00

bench_jagged_causal_local.sh

Move all test and bench scripts to folder scripts

2025-06-06 08:22:38 +00:00

bench_jagged_causal_mattn0_full0.sh

Align the -seqlens=xxx in the mattn0_full0 and mattn256_full256 scripts with the required benchmarks

2025-06-18 16:02:04 +00:00

bench_jagged_causal_mattn256_full256.sh

Align the -seqlens=xxx in the mattn0_full0 and mattn256_full256 scripts with the required benchmarks

2025-06-18 16:02:04 +00:00

bench_jagged_causal.sh

Move all test and bench scripts to folder scripts

2025-06-06 08:22:38 +00:00

benchmark_hstu_attention.sh

Move all test and bench scripts to folder scripts

2025-06-06 08:22:38 +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_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_softmax_attention.sh

Add support of softmax in hstu attention

2025-10-20 14:26:55 +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

2025-08-12 03:06:35 +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

2025-08-12 03:06:35 +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

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