@@ -1976,10 +1976,13 @@ def setUp(self):
19761976 def init_shape (self ):
19771977 self .shape = [10 , 12 ]
19781978
1979+ def test_check_output (self ):
1980+ self .check_output (check_pir = True )
1981+
19791982 def test_check_grad (self ):
19801983 if self .dtype == np .float16 :
19811984 return
1982- self .check_grad (['X' ], 'Out' )
1985+ self .check_grad (['X' ], 'Out' , check_pir = True )
19831986
19841987
19851988class TestTan_float32 (TestTan ):
@@ -2020,6 +2023,7 @@ def test_dygraph_api(self):
20202023 out_ref = np .tan (self .x_np )
20212024 np .testing .assert_allclose (out_ref , out_test .numpy (), rtol = 1e-05 )
20222025
2026+ @test_with_pir_api
20232027 def test_static_api (self ):
20242028 with static_guard ():
20252029 with paddle .static .program_guard (paddle .static .Program ()):
@@ -4815,7 +4819,7 @@ def test_check_grad(self):
48154819 check_pir = True ,
48164820)
48174821create_test_act_fp16_class (TestCos , check_pir = True )
4818- create_test_act_fp16_class (TestTan )
4822+ create_test_act_fp16_class (TestTan , check_pir = True )
48194823create_test_act_fp16_class (TestCosh )
48204824create_test_act_fp16_class (TestAcos , check_pir = True )
48214825create_test_act_fp16_class (TestSin , check_pir = True )
@@ -4976,7 +4980,7 @@ def test_check_grad(self):
49764980 TestFloor , grad_check = False , check_prim = True , check_pir = True
49774981)
49784982create_test_act_bf16_class (TestCos , check_pir = True )
4979- create_test_act_bf16_class (TestTan )
4983+ create_test_act_bf16_class (TestTan , check_pir = True )
49804984create_test_act_bf16_class (TestCosh )
49814985create_test_act_bf16_class (TestAcos , check_pir = True )
49824986create_test_act_bf16_class (TestSin , check_pir = True )
0 commit comments