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

Signed-off-by: Liangfu Chen <[email protected]>
Co-authored-by: George Novack <[email protected]>
Co-authored-by: Aoyu Zhang <[email protected]>

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