We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d958b4d commit e56e9dbCopy full SHA for e56e9db
torchrec/modules/tests/test_hash_mc_modules.py
@@ -694,7 +694,8 @@ def test_dynamically_switch_inference_training_mode(self) -> None:
694
torch.cuda.device_count() < 1,
695
"Not enough GPUs, this test requires at least two GPUs",
696
)
697
- def test_zch_hash_disable_fallback(self) -> None:
+ def test_zch_hash_disable_fallback_disabled_in_oss_compatibility(self) -> None:
698
+ # TODO: fix this flaky test in OSS.
699
m = HashZchManagedCollisionModule(
700
zch_size=30,
701
device=torch.device("cuda"),
0 commit comments