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

…

backend.py

…

test_basic_correctness.py

[Misc] Split up pooling tasks (#10820)

2024-12-11 01:28:00 -08:00

test_full_graph.py

…

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

…

test_wrapper.py

…

utils.py

[bugfix] fix full graph tests (#10581)

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