[Enhancement] Add PyTorch1.11.0 CI

.github/workflows/build.yml

@@ -217,12 +217,14 @@ jobs:
     strategy:
       matrix:
         python-version: [3.7]
-        torch: [1.9.0+cu102, 1.10.0+cu102]
+        torch: [1.9.0+cu102, 1.10.0+cu102, 1.11.0+cu102]
         include:
           - torch: 1.9.0+cu102
             torchvision: 0.10.0+cu102
           - torch: 1.10.0+cu102
             torchvision: 0.11.0+cu102
+          - torch: 1.11.0+cu102
+            torchvision: 0.12.0+cu102
           - python-version: 3.6
             torch: 1.9.0+cu102
             torchvision: 0.10.0+cu102

tests/test_parallel.py

@@ -31,11 +31,15 @@ def __init__(self):
         def forward(self, x):
             return self.conv(x)
 
-    # _verify_model_across_ranks is added in torch1.9.0 so we should check
-    # whether _verify_model_across_ranks is the member of torch.distributed
-    # before mocking
+    # _verify_model_across_ranks is added in torch1.9.0,
+    # _verify_params_across_processes is added in torch1.11.0,
+    # so we should check whether _verify_model_across_ranks
+    # and _verify_params_across_processes are the member of
+    # torch.distributed before mocking
     if hasattr(torch.distributed, '_verify_model_across_ranks'):
         torch.distributed._verify_model_across_ranks = mock
+    if hasattr(torch.distributed, '_verify_params_across_processes'):
+        torch.distributed._verify_params_across_processes = mock
 
     model = Model()
     assert not is_module_wrapper(model)