Skipped unit tests in test_fsdp_sharded_grad_scaler.py

BLOrange-AMD · web-flow · commit eb38990f1ed6 · 2025-03-10T00:08:43.000-05:00
diff --git a/test/distributed/fsdp/test_fsdp_sharded_grad_scaler.py b/test/distributed/fsdp/test_fsdp_sharded_grad_scaler.py
@@ -36,6 +36,8 @@
     run_tests,
     TEST_WITH_DEV_DBG_ASAN,
     TestCase,
+    NAVI_ARCH,
+    skipIfRocmArch,
 )
 
 
@@ -236,6 +238,7 @@ def _build_model_and_optim(
         return model, optim, ref_model, ref_optim
 
     @skip_if_lt_x_gpu(2)
+    @skipIfRocmArch(NAVI_ARCH)
     def test_sharded_grad_scaler_found_inf(self):
         self.run_subtests(
             {

Original file line number	Diff line number	Diff line change
`@@ -36,6 +36,8 @@`
`36`	`36`	`run_tests,`
`37`	`37`	`TEST_WITH_DEV_DBG_ASAN,`
`38`	`38`	`TestCase,`
	`39`	`+ NAVI_ARCH,`
	`40`	`+ skipIfRocmArch,`
`39`	`41`	`)`
`40`	`42`
`41`	`43`
`@@ -236,6 +238,7 @@ def _build_model_and_optim(`
`236`	`238`	`return model, optim, ref_model, ref_optim`
`237`	`239`
`238`	`240`	`@skip_if_lt_x_gpu(2)`
	`241`	`+ @skipIfRocmArch(NAVI_ARCH)`
`239`	`242`	`def test_sharded_grad_scaler_found_inf(self):`
`240`	`243`	`self.run_subtests(`
`241`	`244`	`{`