[Dy2St][PIR] Enable 2 AMP composite tests (#62810)

SigureMo · web-flow · commit badfbafa8d7a · 2024-03-18T17:27:45.000+08:00
diff --git a/test/dygraph_to_static/test_resnet_amp.py b/test/dygraph_to_static/test_resnet_amp.py
@@ -20,7 +20,6 @@
     Dy2StTestBase,
     enable_to_static_guard,
     test_default_and_pir,
-    test_default_mode_only,
 )
 from test_resnet import SEED, ResNet, optimizer_setting
 
@@ -128,7 +127,7 @@ def test_resnet(self):
             err_msg=f'static_loss: {static_loss} \n dygraph_loss: {dygraph_loss}',
         )
 
-    @test_default_mode_only
+    @test_default_and_pir
     def test_resnet_composite(self):
         core._set_prim_backward_enabled(True)
         static_loss = self.train(to_static=True)
diff --git a/test/dygraph_to_static/test_resnet_pure_fp16.py b/test/dygraph_to_static/test_resnet_pure_fp16.py
@@ -20,7 +20,6 @@
     Dy2StTestBase,
     enable_to_static_guard,
     test_default_and_pir,
-    test_default_mode_only,
 )
 from test_resnet import SEED, ResNet, optimizer_setting
 
@@ -138,7 +137,7 @@ def test_resnet(self):
                 err_msg=f'static_loss: {static_loss} \n dygraph_loss: {dygraph_loss}',
             )
 
-    @test_default_mode_only
+    @test_default_and_pir
     def test_resnet_composite(self):
         if base.is_compiled_with_cuda():
             core._set_prim_backward_enabled(True)

Original file line number	Diff line number	Diff line change
`@@ -20,7 +20,6 @@`
`20`	`20`	`Dy2StTestBase,`
`21`	`21`	`enable_to_static_guard,`
`22`	`22`	`test_default_and_pir,`
`23`		`- test_default_mode_only,`
`24`	`23`	`)`
`25`	`24`	`from test_resnet import SEED, ResNet, optimizer_setting`
`26`	`25`
`@@ -128,7 +127,7 @@ def test_resnet(self):`
`128`	`127`	`err_msg=f'static_loss: {static_loss} \n dygraph_loss: {dygraph_loss}',`
`129`	`128`	`)`
`130`	`129`
`131`		`- @test_default_mode_only`
	`130`	`+ @test_default_and_pir`
`132`	`131`	`def test_resnet_composite(self):`
`133`	`132`	`core._set_prim_backward_enabled(True)`
`134`	`133`	`static_loss = self.train(to_static=True)`