youkaichaoandGitHub 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

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

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

piecewise_compilation_config.json

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

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

test_simple.py

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

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

test_toy_llama.py

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

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