Fix OSS test (#3012)

root · facebook-github-bot · commit 83f8beaa7324 · 2025-06-02T14:47:09.000-07:00
Summary: GPU CI tests were failing on this assertion since 10 months. Working test run https://github.com/pytorch/torchrec/actions/runs/15314133743 Pull Request resolved: #3012 Test Plan: Before: {F1978814280} After: {F1978814318} Reviewed By: TroyGarden Differential Revision: D75571603 Pulled By: kausv fbshipit-source-id: 75c057295f2febbf7626b8ea2f0a589795b2da6c
diff --git a/torchrec/distributed/tests/test_mc_embedding.py b/torchrec/distributed/tests/test_mc_embedding.py
@@ -1124,7 +1124,7 @@ def test_sharding_zch_mc_ec_dedup_input_error(self, backend: str) -> None:
                     )
                 ),
                 backend=backend,
-                input_hash_size=(2**62) - 1 + 10,
+                input_hash_size=(2**52) - 1 + 10,
             ),
         except AssertionError as e:
             self.assertTrue("0 != 1" in str(e))

Original file line number	Diff line number	Diff line change
`@@ -1124,7 +1124,7 @@ def test_sharding_zch_mc_ec_dedup_input_error(self, backend: str) -> None:`
`1124`	`1124`	`)`
`1125`	`1125`	`),`
`1126`	`1126`	`backend=backend,`
`1127`		`- input_hash_size=(2**62) - 1 + 10,`
	`1127`	`+ input_hash_size=(2**52) - 1 + 10,`
`1128`	`1128`	`),`
`1129`	`1129`	`except AssertionError as e:`
`1130`	`1130`	`self.assertTrue("0 != 1" in str(e))`