youkaichaoandGitHub 96e0c9cbbd [torch.compile] directly register custom op (#9896) ...

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

2024-10-31 21:56:09 -07:00

..

piecewise

[torch.compile] directly register custom op (#9896)

2024-10-31 21:56:09 -07:00

__init__.py

[torch.compile] register allreduce operations as custom ops (#8526)

2024-09-16 22:57:57 -07:00

test_basic_correctness.py

[torch.compile] upgrade tests (#9858)

2024-10-30 16:34:22 -07:00

test_full_graph.py

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

2024-10-29 23:03:49 -07: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