youkaichaoandGitHub 0cd3d9717e [7/N] torch.compile, reduce compilation time (#10460) ...

Signed-off-by: youkaichao <youkaichao@gmail.com>

2024-11-20 11:20:38 -08:00

..

piecewise

[7/N] torch.compile, reduce compilation time (#10460)

2024-11-20 11:20:38 -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