Add test owner to distributed files starting with test_ (#66797)

Summary:
Action based on https://github.com/pytorch/pytorch/issues/66232

cc pietern mrshenli pritamdamania87 zhaojuanmao satgera rohan-varma gqchen aazzolini osalpekar jiayisuse SciPioneer H-Huang

Pull Request resolved: https://github.com/pytorch/pytorch/pull/66797

Reviewed By: gchanan

Differential Revision: D31761389

Pulled By: janeyx99

fbshipit-source-id: c27c9ab4acec1eb71d5edd4538cd113b770dfc6c

.github/workflows/lint.yml

@@ -102,6 +102,10 @@ jobs:
         if: always()
         run: |
           (! git --no-pager grep -I -no $'cudaStreamSynchronize' --  ./aten ./c10 ':(exclude)aten/src/ATen/test' ':(exclude)c10/cuda/CUDAFunctions.h' || (echo "The above files call raw cuda APIs directly; please use at::cuda wrappers instead"; false))
+      - name: Ensure all test files have header containing ownership information
+        if: always()
+        run: |
+          (! git grep -L "# Owner(s): \[" test/distributed/**/test_*.py) || (printf "The above test files are missing a comment header with ownership information; please add the following line\n\n# Owner(s): [\"<owner: label>\"]\n\nto the top of each test file. The owner should be an existing pytorch/pytorch label."; false)
 
   clang-format:
     runs-on: ubuntu-18.04

test/distributed/_sharded_tensor/ops/test_linear.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 import sys
 import torch
 

test/distributed/_sharded_tensor/test_sharded_tensor.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 import math
 import io
 import itertools

test/distributed/_sharding_spec/test_sharding_spec.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 import torch
 from torch.testing._internal.common_utils import TestCase
 from torch.distributed._sharding_spec import (

test/distributed/algorithms/ddp_comm_hooks/test_ddp_hooks.py

@@ -1,3 +1,4 @@
+# Owner(s): ["oncall: distributed"]
 
 import os
 import sys

test/distributed/algorithms/quantization/test_quantization.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 import torch
 import os
 import torch.cuda

test/distributed/algorithms/test_join.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 import contextlib
 import os
 import sys

test/distributed/bin/test_script.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
+# Owner(s): ["oncall: distributed"]
 
 # Copyright (c) Facebook, Inc. and its affiliates.
 # All rights reserved.

test/distributed/fsdp/test_fsdp.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 import functools
 import sys
 

test/distributed/nn/jit/test_instantiator.py

@@ -1,4 +1,6 @@
 #!/usr/bin/env python3
+# Owner(s): ["oncall: distributed"]
+
 import pathlib
 import sys
 import unittest

test/distributed/optim/test_zero_redundancy_optimizer.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 # Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
 #
 # This source code is licensed under the BSD license found in the

test/distributed/pipeline/sync/test_balance.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 # Copyright 2019 Kakao Brain
 #
 # Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.

test/distributed/pipeline/sync/test_bugs.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 # Copyright 2019 Kakao Brain
 #
 # Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.

test/distributed/pipeline/sync/test_checkpoint.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 # Copyright 2019 Kakao Brain
 #
 # Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.

test/distributed/pipeline/sync/test_copy.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 # Copyright 2019 Kakao Brain
 #
 # Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.

test/distributed/pipeline/sync/test_deferred_batch_norm.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 # Copyright 2019 Kakao Brain
 #
 # Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.

test/distributed/pipeline/sync/test_dependency.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 # Copyright 2019 Kakao Brain
 #
 # Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.

test/distributed/pipeline/sync/test_inplace.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 # Copyright 2019 Kakao Brain
 #
 # Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.

test/distributed/pipeline/sync/test_microbatch.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 # Copyright 2019 Kakao Brain
 #
 # Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.

test/distributed/pipeline/sync/test_phony.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 # Copyright 2019 Kakao Brain
 #
 # Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.

test/distributed/pipeline/sync/test_pipe.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 # Copyright 2019 Kakao Brain
 #
 # Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.

test/distributed/pipeline/sync/test_pipeline.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 # Copyright 2019 Kakao Brain
 #
 # Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.

test/distributed/pipeline/sync/test_stream.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 # Copyright 2019 Kakao Brain
 #
 # Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.

test/distributed/pipeline/sync/test_transparency.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 # Copyright 2019 Kakao Brain
 #
 # Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.

test/distributed/pipeline/sync/test_worker.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 # Copyright 2019 Kakao Brain
 #
 # Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.

test/distributed/rpc/cuda/test_tensorpipe_agent.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
+# Owner(s): ["oncall: distributed"]
 
 import sys
 

test/distributed/rpc/test_faulty_agent.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
+# Owner(s): ["oncall: distributed"]
 
 import sys
 

test/distributed/rpc/test_tensorpipe_agent.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
+# Owner(s): ["oncall: distributed"]
 
 import sys
 

test/distributed/test_c10d_common.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 import copy
 import os
 import sys

test/distributed/test_c10d_gloo.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 import copy
 import logging
 import math

test/distributed/test_c10d_nccl.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 import copy
 import math
 import os

test/distributed/test_c10d_spawn.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 import sys
 import tempfile
 

test/distributed/test_c10d_spawn_gloo.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 import copy
 import os
 import sys

test/distributed/test_c10d_spawn_nccl.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 import sys
 import test_c10d_spawn
 import torch

test/distributed/test_data_parallel.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 import contextlib
 import io
 from copy import deepcopy

test/distributed/test_distributed_spawn.py

@@ -1,3 +1,4 @@
+# Owner(s): ["oncall: distributed"]
 
 import os
 import sys

test/distributed/test_jit_c10d.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 import tempfile
 import sys
 import torch

test/distributed/test_launcher.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 import os
 import sys
 from contextlib import closing

test/distributed/test_nccl.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 import sys
 import torch
 import torch.cuda.nccl as nccl

test/distributed/test_pg_wrapper.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 import os
 import sys
 from datetime import timedelta

test/distributed/test_store.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 import os
 import random
 import sys

test/test_functional_optim.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: distributed"]
+
 import torch
 import torch.nn as nn
 import torch.nn.functional as F