diff --git a/torchrec/distributed/quant_embeddingbag.py b/torchrec/distributed/quant_embeddingbag.py index 2d724db7b..366352180 100644 --- 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