vllm/tests/compile
youkaichao 803f37eaaa
[6/N] torch.compile rollout to users (#10437)
Signed-off-by: youkaichao <youkaichao@gmail.com>
2024-11-19 10:09:03 -08:00
..
piecewise [6/N] torch.compile rollout to users (#10437) 2024-11-19 10:09:03 -08:00
__init__.py [torch.compile] register allreduce operations as custom ops (#8526) 2024-09-16 22:57:57 -07:00
backend.py [torch.compile] Fuse RMSNorm with quant (#9138) 2024-11-08 21:20:08 +00:00
test_basic_correctness.py [6/N] torch.compile rollout to users (#10437) 2024-11-19 10:09:03 -08:00
test_full_graph.py [2/N][torch.compile] make compilation cfg part of vllm cfg (#10383) 2024-11-16 18:02:14 -08:00
test_fusion.py [2/N][torch.compile] make compilation cfg part of vllm cfg (#10383) 2024-11-16 18:02:14 -08:00
test_wrapper.py [2/N][torch.compile] make compilation cfg part of vllm cfg (#10383) 2024-11-16 18:02:14 -08:00
utils.py [6/N] torch.compile rollout to users (#10437) 2024-11-19 10:09:03 -08:00