Skip to content

Commit 7c6e45b

Browse files
TroyGardenmeta-codesync[bot]
authored andcommitted
disable flasky test in github workflow (#3528)
Summary: Pull Request resolved: #3528 Disabled a flaky test in the TorchRec sequence model parallel test suite to prevent intermittent failures in the GitHub workflow CI pipeline. The test was causing unreliable builds and needed to be disabled until the underlying issue can be properly investigated and resolved. NOTE: The first CUDA IMA error will cause all following cuda-related tests to fail {F1983370186} Reviewed By: spmex Differential Revision: D86021612 fbshipit-source-id: 4c8c50e8215a009213089e405590e21c5b0060bc
1 parent da91c05 commit 7c6e45b

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

torchrec/distributed/tests/test_sequence_model_parallel.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -395,7 +395,9 @@ class DedupIndicesWeightAccumulationTest(unittest.TestCase):
395395
not torch.cuda.is_available(),
396396
"Not enough GPUs, this test requires at least one GPU",
397397
)
398-
def test_dedup_indices_weight_accumulation(self) -> None:
398+
def test_dedup_indices_weight_accumulation_disabled_in_oss_compatibility(
399+
self,
400+
) -> None:
399401
"""
400402
Test the _dedup_indices method to ensure weight accumulation works correctly
401403
with the new scatter_add_along_first_dim implementation.

0 commit comments

Comments
 (0)