Liangfu ChenandGitHub d2b58ca203 [Neuron][kernel] Fuse kv cache into a single tensor (#15911) ...

Signed-off-by: Liangfu Chen <liangfc@amazon.com>

2025-04-03 09:51:32 -07:00

..

test_activation.py

[Neuron] flatten test parameterization for neuron attention kernels (#14712)

2025-03-13 20:46:56 -07:00

test_block_table.py

[Misc] Replace os environ to monkeypatch in test suite (#14516)

2025-03-16 20:35:57 -07:00

test_cache.py

[Neuron][kernel] Fuse kv cache into a single tensor (#15911)

2025-04-03 09:51:32 -07:00

test_layernorm.py

[Neuron] flatten test parameterization for neuron attention kernels (#14712)

2025-03-13 20:46:56 -07:00

test_logits_processor.py

[Neuron] flatten test parameterization for neuron attention kernels (#14712)

2025-03-13 20:46:56 -07:00

test_prefix_prefill.py

[Neuron][kernel] Fuse kv cache into a single tensor (#15911)

2025-04-03 09:51:32 -07:00

test_rotary_embedding.py

[Neuron] flatten test parameterization for neuron attention kernels (#14712)

2025-03-13 20:46:56 -07:00