Antoni Baum a10d3056da

[Core] Set linear_weights directly on the layer (#3977)

2024-04-11 16:35:51 -04:00

..

allclose_default.py

[ROCm] Fix some kernels failed unit tests (#2498)

2024-02-05 14:25:36 -08:00

conftest.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

test_activation.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

test_attention.py

[Misc] Add indirection layer for custom ops (#3913)

2024-04-10 20:26:07 -07:00

test_cache.py

[Misc] Add indirection layer for custom ops (#3913)

2024-04-10 20:26:07 -07:00

test_layernorm.py

[Kernel] Layernorm performance optimization (#3662)

2024-03-30 14:26:38 -07:00

test_moe.py

[Core] Set linear_weights directly on the layer (#3977)

2024-04-11 16:35:51 -04:00

test_pos_encoding.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

test_prefix_prefill.py

feat(benchmarks): Add Prefix Caching Benchmark to Serving Benchmark (#3277)

2024-03-27 13:39:26 -07:00

test_rand.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

test_sampler.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00