youkaichao 32176fee73

[torch.compile] support moe models (#9632) ...

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

2024-10-27 21:58:04 -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] support moe models (#9632)

2024-10-27 21:58:04 -07:00

test_full_graph.py

[torch.compile] integration with compilation control (#9058)

2024-10-10 12:39:36 -07:00

test_wrapper.py

[tpu][misc] fix typo (#8260)

2024-09-06 22:40:46 -07:00

utils.py

[Hardware][ROCM] using current_platform.is_rocm (#9642)

2024-10-28 04:07:00 +00:00