File tree Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -689,12 +689,13 @@ def test_dynamically_switch_inference_training_mode(self) -> None:
689
689
self .assertTrue (m ._eviction_policy_name is None )
690
690
self .assertTrue (m ._eviction_module is None )
691
691
692
+ # Skipping this test because it is flaky on CI. TODO: T240185573 T240185565 investigate the flakiness and re-enable the test.
692
693
# Pyre-ignore [56]: Pyre was not able to infer the type of argument `torch.cuda.device_count() < 1` to decorator factory `unittest.skipIf`
693
694
@unittest .skipIf (
694
695
torch .cuda .device_count () < 1 ,
695
696
"Not enough GPUs, this test requires at least two GPUs" ,
696
697
)
697
- def test_zch_hash_disable_fallback (self ) -> None :
698
+ def test_zch_hash_disable_fallback_disabled_in_oss_compatatibility (self ) -> None :
698
699
m = HashZchManagedCollisionModule (
699
700
zch_size = 30 ,
700
701
device = torch .device ("cuda" ),
You can’t perform that action at this time.
0 commit comments