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