youkaichao ff5ed6e1bc

[torch.compile] rework compile control with piecewise cudagraph (#9715) ...

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

2024-10-29 23:03:49 -07:00

..

__init__.py

[Bugfix] Fix multi nodes TP+PP for XPU (#8884)

2024-10-29 21:34:45 -07:00

cpu.py

[CI/Build] Add test decorator for minimum GPU memory (#8925)

2024-09-29 02:50:51 +00:00

cuda.py

[Misc] Upgrade to pytorch 2.5 (#9588)

2024-10-27 09:44:24 +00:00

interface.py

[Hardware] using current_platform.seed_everything (#9785)

2024-10-29 14:47:44 +00:00

neuron.py

[Neuron] [Bugfix] Fix neuron startup (#9374)

2024-10-22 12:51:41 +00:00

openvino.py

[Hardware][openvino] is_openvino --> current_platform.is_openvino (#9716)

2024-10-26 10:59:06 +00:00

rocm.py

[CI/Build] Add test decorator for minimum GPU memory (#8925)

2024-09-29 02:50:51 +00:00

tpu.py

[torch.compile] rework compile control with piecewise cudagraph (#9715)

2024-10-29 23:03:49 -07:00

xpu.py

[Bugfix] Fix multi nodes TP+PP for XPU (#8884)

2024-10-29 21:34:45 -07:00