Michael Goin f78c0be80a Fix benchmark_moe.py tuning for CUDA devices (#14164)

2025-03-03 21:11:03 -08:00

..

benchmark_aqlm.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

benchmark_layernorm.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

benchmark_lora.py

Update deprecated Python 3.8 typing (#13971)

2025-03-02 17:34:51 -08:00

benchmark_machete.py

Update deprecated Python 3.8 typing (#13971)

2025-03-02 17:34:51 -08:00

benchmark_marlin.py

Update deprecated Python 3.8 typing (#13971)

2025-03-02 17:34:51 -08:00

benchmark_moe.py

Fix benchmark_moe.py tuning for CUDA devices (#14164)

2025-03-03 21:11:03 -08:00

benchmark_paged_attention.py

[ROCm] Faster Custom Paged Attention kernels (#12348)

2025-03-03 09:24:45 -08:00

benchmark_quant.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

benchmark_rmsnorm.py

Update deprecated Python 3.8 typing (#13971)

2025-03-02 17:34:51 -08:00

benchmark_rope.py

Update deprecated Python 3.8 typing (#13971)

2025-03-02 17:34:51 -08:00

benchmark_shapes.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

graph_machete_bench.py

Update deprecated Python 3.8 typing (#13971)

2025-03-02 17:34:51 -08:00

requirements.txt

[Kernel] (2/N) Machete - Integrate into CompressedTensorsWNA16 and GPTQMarlin (#7701)

2024-09-23 13:46:26 -04:00

utils.py

Update deprecated Python 3.8 typing (#13971)

2025-03-02 17:34:51 -08:00

weight_shapes.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00