Skip to content

Commit a784f87

Browse files
committed
Skip one more scheduled GPU integration tests
1 parent 21e3372 commit a784f87

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

tests/integration/train_tests.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -505,6 +505,7 @@ def test_tpu_zero1_gradient_accumulation(self):
505505
@pytest.mark.integration_test
506506
@pytest.mark.gpu_only
507507
@pytest.mark.scheduled_only
508+
@pytest.mark.skip(reason="b/489133823. Previously transient in b/462548581.")
508509
def test_gpu_zero1_gradient_accumulation(self):
509510
os.environ["NVTE_FUSED_ATTN"] = "1" # Enable fused attention
510511
zero1_ga = [ # tests Zero-1 optimizer sharding with gradient accumulation

0 commit comments

Comments
 (0)