Luka Govedič 30870b4f66 [torch.compile] Dynamic fp8 + rms_norm fusion (#10906) ...

Signed-off-by: luka <luka@neuralmagic.com>
Co-authored-by: Varun Sundar Rabindranath <varun@neuralmagic.com>

2024-12-13 03:19:23 +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

[Misc] Split up pooling tasks (#10820)

2024-12-11 01:28:00 -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_functionalization.py

[torch.compile] Dynamic fp8 + rms_norm fusion (#10906)

2024-12-13 03:19:23 +00:00

test_fusion.py

[torch.compile] Dynamic fp8 + rms_norm fusion (#10906)

2024-12-13 03:19:23 +00: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