Fix failed unit test test_sharder_different_world_sizes_for_qec (#1954)

Summary: Pull Request resolved: #1954 Differential Revision: D56970252
pytorch · May 6, 2024 · 6a0f6cd · 6a0f6cd
1 parent 9590907
commit 6a0f6cd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torchrec/distributed/quant_embedding.py b/torchrec/distributed/quant_embedding.py
@@ -599,7 +599,7 @@ def input_dist(
         features: KeyedJaggedTensor,
     ) -> ListOfKJTList:
         if self._has_uninitialized_input_dist:
-            self._intput_dist = ShardedQuantEcInputDist(
+            self._input_dist = ShardedQuantEcInputDist(
                 input_feature_names=features.keys() if features is not None else [],
                 sharding_type_to_sharding=self._sharding_type_to_sharding,
                 device=self._device,
@@ -614,7 +614,7 @@ def input_dist(
             input_dist_result_list,
             features_by_sharding,
             unbucketize_permute_tensor_list,
-        ) = self._intput_dist(features)
+        ) = self._input_dist(features)
 
         with torch.no_grad():
             for i in range(len(self._sharding_type_to_sharding)):