Liangfu Chen f75aa72732

[Neuron] Add custom_ops for neuron backend (#13246) ...

Signed-off-by: Liangfu Chen <liangfc@amazon.com>
Co-authored-by: George Novack <gnovack@amazon.com>
Co-authored-by: Aoyu Zhang <aoyuzhan@amazon.com>

2025-02-25 11:47:49 -08:00

..

test_activation.py

[Neuron] Add custom_ops for neuron backend (#13246)

2025-02-25 11:47:49 -08:00

test_block_table.py

[Neuron][Kernel] Vectorize KV cache load in FlashPagedAttention to maximize DMA bandwidth (#13245)

2025-02-20 17:45:45 -08:00

test_layernorm.py

[Neuron] Add custom_ops for neuron backend (#13246)

2025-02-25 11:47:49 -08:00

test_logits_processor.py

[Neuron] Add custom_ops for neuron backend (#13246)

2025-02-25 11:47:49 -08:00

test_prefix_prefill.py

[Neuron] Add custom_ops for neuron backend (#13246)

2025-02-25 11:47:49 -08:00

test_rotary_embedding.py

[Neuron] Add custom_ops for neuron backend (#13246)

2025-02-25 11:47:49 -08:00