AlpinDale 239a8cae25 torch.compile: register all-reduce operations as custom ops (#1050) | vor 1 Woche | |
---|---|---|
.. | ||
__init__.py | 239a8cae25 torch.compile: register all-reduce operations as custom ops (#1050) | vor 1 Woche |
test_full_graph.py | 239a8cae25 torch.compile: register all-reduce operations as custom ops (#1050) | vor 1 Woche |
test_wrapper.py | 0e5cf7f840 tpu: avoid dynamo guard eval overhead (#949) | vor 2 Wochen |