vllm/tests/compile
Luka Govedič 4f93dfe952
[torch.compile] Fuse RMSNorm with quant (#9138)
Signed-off-by: luka <luka@neuralmagic.com>
Co-authored-by: youkaichao <youkaichao@126.com>
2024-11-08 21:20:08 +00:00
..
piecewise [CI/Build] drop support for Python 3.8 EOL (#8464) 2024-11-06 07:11:55 +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] Fuse RMSNorm with quant (#9138) 2024-11-08 21:20:08 +00:00
test_basic_correctness.py [torch.compile] rework test plans (#9866) 2024-10-31 22:20:17 -07:00
test_full_graph.py [torch.compile] rework compile control with piecewise cudagraph (#9715) 2024-10-29 23:03:49 -07:00
test_fusion.py [torch.compile] Fuse RMSNorm with quant (#9138) 2024-11-08 21:20:08 +00:00
test_wrapper.py [tpu][misc] fix typo (#8260) 2024-09-06 22:40:46 -07:00
utils.py [torch.compile] rework compile control with piecewise cudagraph (#9715) 2024-10-29 23:03:49 -07:00