diff --git a/python/paddle/v2/framework/tests/op_test.py b/python/paddle/v2/framework/tests/op_test.py index 9b2f10fdf95803cf3613b3e06e547235c8a105cb..3f8e1236ff7721519a7bcb7489757202773c039d 100644 --- a/python/paddle/v2/framework/tests/op_test.py +++ b/python/paddle/v2/framework/tests/op_test.py @@ -256,7 +256,7 @@ class OpTest(unittest.TestCase): ] places = [core.CPUPlace()] - if core.is_compile_gpu() and op.support_gpu(): + if core.is_compile_gpu() and self.op.support_gpu(): places.append(core.GPUPlace(0)) for place in places: