sharding debug

fbshipit-source-id: 4a0b3c3de51d347f46f303bf37e7d64688d41d98
s4ayub · Feb 1, 2024 · 5173e5d · 5173e5d
1 parent 2d52760
commit 5173e5d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchrec/distributed/quant_embeddingbag.py b/torchrec/distributed/quant_embeddingbag.py
@@ -407,6 +407,7 @@ def __init__(
         assert feature_processor is not None
         device_type: str = self._device.type if self._device is not None else "cuda"
         self.feature_processors_per_rank: nn.ModuleList = torch.nn.ModuleList()
+        breakpoint()
         for i in range(env.world_size):
             self.feature_processors_per_rank.append(
                 feature_processor