Skip to content

Commit efc2186

Browse files
emlinmeta-codesync[bot]
authored andcommitted
fix cpu ci test (#3524)
Summary: Pull Request resolved: #3524 for non GPU environment, skip the ut Reviewed By: kausv Differential Revision: D86475425 fbshipit-source-id: 4033dccce93e9b3b1241e1b6171ce9be985474b7
1 parent 6bc2d34 commit efc2186

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

torchrec/distributed/tests/test_sequence_model_parallel.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -390,7 +390,11 @@ class DedupIndicesWeightAccumulationTest(unittest.TestCase):
390390
This tests the correctness of the new scatter_add_along_first_dim implementation.
391391
"""
392392

393-
# to be deleted
393+
# pyre-fixme[56]: Pyre was not able to infer the type of argument
394+
@unittest.skipIf(
395+
not torch.cuda.is_available(),
396+
"Not enough GPUs, this test requires at least one GPU",
397+
)
394398
def test_dedup_indices_weight_accumulation(self) -> None:
395399
"""
396400
Test the _dedup_indices method to ensure weight accumulation works correctly

0 commit comments

Comments
 (0)