Skip to content

Commit 3eba028

Browse files
nipung90meta-codesync[bot]
authored andcommitted
Disable test_zch_hash_disable_fallback in oss because of flakiness (#3426)
Summary: Pull Request resolved: #3426 Disable test_zch_hash_disable_fallback in oss because of flakiness Reviewed By: aliafzal, spmex Differential Revision: D83788664 fbshipit-source-id: f91f315f23bba0b13de8446e520f3371890089a4
1 parent 107678b commit 3eba028

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

torchrec/modules/tests/test_hash_mc_modules.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -689,12 +689,13 @@ def test_dynamically_switch_inference_training_mode(self) -> None:
689689
self.assertTrue(m._eviction_policy_name is None)
690690
self.assertTrue(m._eviction_module is None)
691691

692+
# Skipping this test because it is flaky on CI. TODO: T240185573 T240185565 investigate the flakiness and re-enable the test.
692693
# Pyre-ignore [56]: Pyre was not able to infer the type of argument `torch.cuda.device_count() < 1` to decorator factory `unittest.skipIf`
693694
@unittest.skipIf(
694695
torch.cuda.device_count() < 1,
695696
"Not enough GPUs, this test requires at least two GPUs",
696697
)
697-
def test_zch_hash_disable_fallback(self) -> None:
698+
def test_zch_hash_disable_fallback_disabled_in_oss_compatatibility(self) -> None:
698699
m = HashZchManagedCollisionModule(
699700
zch_size=30,
700701
device=torch.device("cuda"),

0 commit comments

Comments
 (0)