diff --git a/python/paddle/fluid/tests/unittests/mkldnn/test_flags_mkldnn_ops_on_off.py b/python/paddle/fluid/tests/unittests/mkldnn/test_flags_mkldnn_ops_on_off.py index 52dc71c0ed84e5a2b0e3c85c3cbf9e16a71ca240..8052dc8f1e7b8211dff478fe43b584a29b8fac86 100644 --- a/python/paddle/fluid/tests/unittests/mkldnn/test_flags_mkldnn_ops_on_off.py +++ b/python/paddle/fluid/tests/unittests/mkldnn/test_flags_mkldnn_ops_on_off.py @@ -31,9 +31,9 @@ class TestFlagsUseMkldnn(unittest.TestCase): self.env[str("DNNL_VERBOSE")] = str("1") self.env[str("FLAGS_use_mkldnn")] = str("1") - self.relu_regex = r"^dnnl_verbose,exec,cpu,eltwise,.+alg:eltwise_relu alpha:0 beta:0,10x20x20" - self.ew_add_regex = r"^dnnl_verbose,exec,cpu,binary.+alg:binary_add,10x20x30:10x20x30 10x20x30" - self.matmul_regex = r"^dnnl_verbose,exec,cpu,matmul,.*b10m20n20k30" + self.relu_regex = b"^dnnl_verbose,exec,cpu,eltwise,.+alg:eltwise_relu alpha:0 beta:0,10x20x20" + self.ew_add_regex = b"^dnnl_verbose,exec,cpu,binary.+alg:binary_add,10x20x30:10x20x30 10x20x30" + self.matmul_regex = b"^dnnl_verbose,exec,cpu,matmul,.*b10m20n20k30" def flags_use_mkl_dnn_common(self, e): cmd = self._python_interp