MKLDNN elementwise_mul: Disable UT for CUDA

test=develop

Author: Sand3r-

python/paddle/fluid/tests/unittests/op_test.py

@@ -362,7 +362,9 @@ def _get_places(self):
             else:
                 return []
         places = [fluid.CPUPlace()]
-        if core.is_compiled_with_cuda() and core.op_support_gpu(self.op_type):
+        cpu_only = self._cpu_only if hasattr(self, '_cpu_only') else False
+        if core.is_compiled_with_cuda() and core.op_support_gpu(self.op_type)\
+           and not cpu_only:
             places.append(core.CUDAPlace(0))
         return places
 

python/paddle/fluid/tests/unittests/test_elementwise_mul_mkldnn_op.py

@@ -34,6 +34,7 @@ def setUp(self):
         super(TestElementwiseMulMKLDNNOp_BroadcastNCHW16c, self).setUp()
         self.attrs["x_data_format"] = "nchw16c"
         self.attrs["y_data_format"] = "nc"
+        self._cpu_only = True
 
     def init_kernel_type(self):
         self.use_mkldnn = True
@@ -66,6 +67,7 @@ def setUp(self):
         super(TestElementwiseMulMKLDNNOp_BroadcastNCHW8c, self).setUp()
         self.attrs["x_data_format"] = "nchw8c"
         self.attrs["y_data_format"] = "nc"
+        self._cpu_only = True
 
     def init_kernel_type(self):
         self.use_mkldnn = True
@@ -119,6 +121,7 @@ def setUp(self):
         super(TestElementwiseMulMKLDNNOp_FallbackNCHW16C, self).setUp()
         self.attrs["x_data_format"] = "nchw16c"
         self.attrs["y_data_format"] = "nchw16c"
+        self._cpu_only = True
 
     def init_kernel_type(self):
         self.use_mkldnn = True
@@ -149,6 +152,7 @@ def setUp(self):
         super(TestElementwiseMulMKLDNNOp_FallbackNoReorders, self).setUp()
         self.attrs["x_data_format"] = "nchw16c"
         self.attrs["y_data_format"] = "nchw16c"
+        self._cpu_only = True
 
     def init_kernel_type(self):
         self.use_mkldnn = True
@@ -178,6 +182,7 @@ def setUp(self):
         super(TestElementwiseMulMKLDNNOp_FallbackWithReorder1, self).setUp()
         self.attrs["x_data_format"] = "nchw"
         self.attrs["y_data_format"] = "nchw16c"
+        self._cpu_only = True
 
     def init_kernel_type(self):
         self.use_mkldnn = True
@@ -207,6 +212,7 @@ def setUp(self):
         super(TestElementwiseMulMKLDNNOp_FallbackWithReorder2, self).setUp()
         self.attrs["x_data_format"] = "nchw16c"
         self.attrs["y_data_format"] = "nchw"
+        self._cpu_only = True
 
     def init_kernel_type(self):
         self.use_mkldnn = True
@@ -235,6 +241,7 @@ def setUp(self):
         super(TestElementwiseMulMKLDNNOp_FallbackNoReorders2, self).setUp()
         self.attrs["x_data_format"] = "nc"
         self.attrs["y_data_format"] = "nc"
+        self._cpu_only = True
 
     def init_kernel_type(self):
         self.use_mkldnn = True