mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
e3c5c369ba
Part of my effort to move everything to pytest and decrease the number of testrunner frameworks in ci Gives xmls but they might look a weird b/c module level tests vs tests in classes. Doesn't give skip/disable test infra because those are tied to classes. (for future ref, could either put tests in classes or move the check_if_enable stuff into a pytest hook) Tested in CI and checked that the same number of tests are run Pull Request resolved: https://github.com/pytorch/pytorch/pull/95659 Approved by: https://github.com/huydhn
37 lines
991 B
Python
37 lines
991 B
Python
# Owner(s): ["oncall: distributed"]
|
|
|
|
# Copyright 2019 Kakao Brain
|
|
#
|
|
# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
|
|
#
|
|
# This source code is licensed under the BSD license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
from torch.distributed.pipeline.sync.pipeline import _clock_cycles
|
|
from torch.testing._internal.common_utils import run_tests
|
|
|
|
|
|
def test_clock_cycles():
|
|
assert list(_clock_cycles(1, 1)) == [[(0, 0)]]
|
|
assert list(_clock_cycles(1, 3)) == [[(0, 0)], [(0, 1)], [(0, 2)]]
|
|
assert list(_clock_cycles(3, 1)) == [[(0, 0)], [(1, 0)], [(2, 0)]]
|
|
|
|
assert list(_clock_cycles(3, 3)) == [
|
|
[(0, 0)],
|
|
[(1, 0), (0, 1)],
|
|
[(2, 0), (1, 1), (0, 2)],
|
|
[(2, 1), (1, 2)],
|
|
[(2, 2)],
|
|
]
|
|
|
|
assert list(_clock_cycles(4, 2)) == [
|
|
[(0, 0)],
|
|
[(1, 0), (0, 1)],
|
|
[(2, 0), (1, 1)],
|
|
[(3, 0), (2, 1)],
|
|
[(3, 1)],
|
|
]
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_tests()
|