fix

Author: co63oc

test/dygraph_to_static/test_build_strategy.py

@@ -77,10 +77,10 @@ def test_resnet(self):
         self.verify_predict()
 
     @test_default_mode_only
-    def test_in_static_mode_onednn(self):
+    def test_in_static_mode_mkldnn(self):
         paddle.set_flags({'FLAGS_use_onednn': True})
         try:
-            if paddle.base.core.is_compiled_with_onednn():
+            if paddle.base.core.is_compiled_with_mkldnn():
                 self.resnet_helper.train(True, self.build_strategy)
         finally:
             paddle.set_flags({'FLAGS_use_onednn': False})

test/dygraph_to_static/test_mnist.py

@@ -173,18 +173,18 @@ def test_mnist_to_static(self):
         )
 
     @test_default_mode_only
-    def test_mnist_declarative_cpu_vs_onednn(self):
+    def test_mnist_declarative_cpu_vs_mkldnn(self):
         dygraph_loss_cpu = self.train_dygraph()
         paddle.set_flags({'FLAGS_use_onednn': True})
         try:
-            dygraph_loss_onednn = self.train_dygraph()
+            dygraph_loss_mkldnn = self.train_dygraph()
         finally:
             paddle.set_flags({'FLAGS_use_onednn': False})
         np.testing.assert_allclose(
             dygraph_loss_cpu,
-            dygraph_loss_onednn,
+            dygraph_loss_mkldnn,
             rtol=1e-05,
-            err_msg=f'cpu dygraph is {dygraph_loss_cpu}\n onednn dygraph is \n{dygraph_loss_onednn}',
+            err_msg=f'cpu dygraph is {dygraph_loss_cpu}\n mkldnn dygraph is \n{dygraph_loss_mkldnn}',
         )
 
     def train(self, to_static=False):

test/dygraph_to_static/test_resnet.py

@@ -474,10 +474,10 @@ def test_resnet_composite(self):
         )
 
     @test_default_mode_only
-    def test_in_static_mode_onednn(self):
+    def test_in_static_mode_mkldnn(self):
         paddle.set_flags({'FLAGS_use_onednn': True})
         try:
-            if paddle.base.core.is_compiled_with_onednn():
+            if paddle.base.core.is_compiled_with_mkldnn():
                 self.train(to_static=True)
         finally:
             paddle.set_flags({'FLAGS_use_onednn': False})