add alias paddle.utils.data.DistributedSampler

Author: algorithm1832

python/paddle/utils/data/__init__.py

@@ -25,6 +25,7 @@
     TensorDataset,
     random_split,
 )
+from .distributed import DistributedSampler
 from .sampler import (
     BatchSampler,
     RandomSampler,
@@ -42,6 +43,7 @@
     'Subset',
     'random_split',
     'BatchSampler',
+    'DistributedSampler',
     'RandomSampler',
     'Sampler',
     'SequentialSampler',

python/paddle/utils/data/distributed.py

@@ -0,0 +1,41 @@
+# Copyright (c) 2026 PaddlePaddle Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from paddle.io import DistributedBatchSampler
+
+if TYPE_CHECKING:
+    from collections.abc import Sized
+
+
+class DistributedSampler(DistributedBatchSampler):
+    def __init__(
+        self,
+        dataset: Sized,
+        num_replicas: int | None = None,
+        rank: int | None = None,
+        shuffle: bool = True,
+        drop_last: bool = False,
+    ) -> None:
+        super().__init__(
+            dataset,
+            1,
+            num_replicas,
+            rank,
+            shuffle,
+            drop_last,
+        )

test/legacy_test/test_api_compatibility_part1.py

@@ -2396,5 +2396,54 @@ def test_Compatibility(self):
         paddle.set_rng_state(new_state=states)
 
 
+# Test DistributedSampler compatibility
+class TestDistributedSamplerAPI(unittest.TestCase):
+    def setUp(self):
+        np.random.seed(2025)
+
+        class SimpleDataset:
+            def __init__(self, size):
+                self.size = size
+
+            def __len__(self):
+                return self.size
+
+        self.dataset = SimpleDataset(100)
+
+    def test_dygraph_Compatibility(self):
+        """Test DistributedSampler as alias for DistributedBatchSampler"""
+        # 1. positional arguments
+        sampler1 = paddle.utils.data.DistributedSampler(
+            self.dataset, 2, 0, False, False
+        )
+        # 2. keyword arguments
+        sampler2 = paddle.utils.data.DistributedSampler(
+            dataset=self.dataset,
+            num_replicas=2,
+            rank=0,
+            shuffle=False,
+            drop_last=False,
+        )
+        # Verify both samplers produce same batches
+        batches1 = list(sampler1)
+        batches2 = list(sampler2)
+        self.assertEqual(len(batches1), len(batches2))
+        for b1, b2 in zip(batches1, batches2):
+            self.assertEqual(b1, b2)
+
+    def test_set_epoch(self):
+        """Test set_epoch method"""
+        sampler = paddle.utils.data.DistributedSampler(
+            self.dataset, num_replicas=2, rank=0, shuffle=True
+        )
+        # Collect samples from epoch 0
+        sampler.set_epoch(0)
+        batches0 = list(sampler)
+        # Collect samples from epoch 1
+        sampler.set_epoch(1)
+        batches1 = list(sampler)
+        self.assertEqual(len(batches0), len(batches1))
+
+
 if __name__ == '__main__':
     unittest.main()