skip rocm

liangel-02 · liangel-02 · commit a19a3edbeb69 · 2025-11-24T15:52:10.000-08:00
diff --git a/tests/integration_tests/features.py b/tests/integration_tests/features.py
@@ -352,6 +352,7 @@ def build_features_test_list() -> list[OverrideDefinitions]:
                     "--parallelism.data_parallel_shard_degree=4",
                     "--activation_checkpoint.mode='full'",
                     "--model.flavor=debugmodel_varlen_attn",
+                    "skip_rocm_test=True",
                 ]
             ],
             "FSDP+VARLEN_ATTN",

Original file line number	Diff line number	Diff line change
`@@ -352,6 +352,7 @@ def build_features_test_list() -> list[OverrideDefinitions]:`
`352`	`352`	`"--parallelism.data_parallel_shard_degree=4",`
`353`	`353`	`"--activation_checkpoint.mode='full'",`
`354`	`354`	`"--model.flavor=debugmodel_varlen_attn",`
	`355`	`+ "skip_rocm_test=True",`
`355`	`356`	`]`
`356`	`357`	`],`
`357`	`358`	`"FSDP+VARLEN_ATTN",`