@@ -70,26 +70,27 @@ def setUp(self):
7070
7171 self .init_dtype ()
7272 np .random .seed (SEED )
73- x = np .random .uniform (1 , 2 , [11 , 17 ]).astype (self .dtype )
74- out = np .power (x , 3 )
73+ x = np .random .uniform (1 , 2 , [3 , 4 ]).astype (self .dtype )
74+ out = np .power (x , 2 )
7575
7676 self .inputs = {'X' : OpTest .np_dtype_to_fluid_dtype (x )}
77- self .attrs = {'factor' : 3 .0 }
77+ self .attrs = {'factor' : 2 .0 }
7878 self .outputs = {'Out' : out }
7979
8080 def set_npu (self ):
8181 self .__class__ .use_npu = True
82+ self .__class__ .no_need_check_grad = True
8283
8384 def init_dtype (self ):
8485 self .dtype = np .float16
8586
8687 def test_check_output (self ):
87- self .check_output_with_place (self .place , check_dygraph = False )
88+ self .check_output_with_place (self .place , check_dygraph = False , atol = 1e-5 )
8889
8990
9091@unittest .skipIf (not paddle .is_compiled_with_npu (),
9192 "core is not compiled with NPU" )
92- class TestSubtractNet (unittest .TestCase ):
93+ class TestPowNet (unittest .TestCase ):
9394 def _test (self , run_npu = True ):
9495 main_prog = paddle .static .Program ()
9596 startup_prog = paddle .static .Program ()
@@ -126,6 +127,7 @@ def _test(self, run_npu=True):
126127 exe = paddle .static .Executor (place )
127128 exe .run (startup_prog )
128129
130+ print ("Start run on {}" .format (place ))
129131 for epoch in range (100 ):
130132
131133 pred_res , loss_res = exe .run (
0 commit comments