youkaichaoandGitHub dc5ce861bf [torch.compile] remove compilation_context and simplify code (#10838) ...

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

2024-12-03 06:19:02 +00:00

..

piecewise

[torch.compile] remove compilation_context and simplify code (#10838)

2024-12-03 06:19:02 +00:00

__init__.py

[torch.compile] register allreduce operations as custom ops (#8526)

2024-09-16 22:57:57 -07:00

backend.py

[torch.compile] Inductor code caching fix (#10273)

2024-11-20 21:44:57 -08:00

test_basic_correctness.py

[torch.compile] support encoder based models (#10613)

2024-11-25 05:24:33 +00: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_functionalization.py

[torch.compile] Inductor code caching fix (#10273)

2024-11-20 21:44:57 -08:00

test_fusion.py

[torch.compile] Inductor code caching fix (#10273)

2024-11-20 21:44:57 -08:00

test_pass_manager.py

[torch.compile] Inductor code caching fix (#10273)

2024-11-20 21:44:57 -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

[bugfix] fix full graph tests (#10581)

2024-11-22 10:02:14 -08:00