Skip to content

Commit ef3e5ca

Browse files
TroyGardenfacebook-github-bot
authored andcommitted
disable a flaky test (#3401)
Summary: Pull Request resolved: #3401 # context * we found this test is flaky in OSS/Github action: [example job](https://github.com/pytorch/torchrec/actions/runs/18104978459/job/51517201520) {F1982344248} * once a `cudaErrorAssert` is triggerred, all the following tests are failing too Reviewed By: nipung90 Differential Revision: D83495599 fbshipit-source-id: c89e6b70878e5067fe1e709a00bc0c2f138b3708
1 parent c3a445e commit ef3e5ca

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

torchrec/modules/tests/test_itep_embedding_modules.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -523,7 +523,9 @@ def mock_forward(
523523
torch.cuda.device_count() <= 1,
524524
"Not enough GPUs, this test requires at least one GPU",
525525
)
526-
def test_error_handling_invalid_iter_tensor_values_cuda(self) -> None:
526+
def test_error_handling_invalid_iter_tensor_values_cuda_disabled_in_oss_compatibility(
527+
self,
528+
) -> None:
527529
"""Test behavior with invalid iter tensor values."""
528530
itep_module = GenericITEPModule(
529531
table_name_to_unpruned_hash_sizes=self._table_name_to_unpruned_hash_sizes,

0 commit comments

Comments
 (0)