added case for skipping dygraph check

PaddlePaddle · Jul 1, 2021 · 890a2e6 · 890a2e6
1 parent ae8dddf
commit 890a2e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/paddle/fluid/tests/unittests/op_test.py b/python/paddle/fluid/tests/unittests/op_test.py
@@ -360,7 +360,7 @@ def try_call_once(self, data_type):
     def is_bfloat16_op(self):
         return self.dtype == np.uint16 or (
             hasattr(self, 'mkldnn_data_type') and
-            getattr(self, 'mkldnn_data_type') is "bfloat16")
+            getattr(self, 'mkldnn_data_type') is "bfloat16") or ('mkldnn_data_type' in self.attrs and self.attrs['mkldnn_data_type'] == 'bfloat16')
 
     def infer_dtype_from_inputs_outputs(self, inputs, outputs):
         def is_np_data(input):