Liangfu ChenandGitHub 53a0cf8b95 [Neuron] trim attention kernel tests to fit trn1.2x instance (#14988) ...

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

2025-03-18 15:05:52 +08: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] flatten test parameterization for neuron attention kernels (#14712)

2025-03-13 20:46:56 -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] trim attention kernel tests to fit trn1.2x instance (#14988)

2025-03-18 15:05:52 +08:00

test_rotary_embedding.py

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

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