diff --git a/python/paddle/fluid/tests/unittests/ngraph/CMakeLists.txt b/python/paddle/fluid/tests/unittests/ngraph/CMakeLists.txt index e9866699a604bc119b45e1b21155694e1e16d396..5ed2d0aa80cd0462d3ac1902a2ec13fc2c1bd844 100644 --- a/python/paddle/fluid/tests/unittests/ngraph/CMakeLists.txt +++ b/python/paddle/fluid/tests/unittests/ngraph/CMakeLists.txt @@ -1,8 +1,6 @@ file(GLOB TEST_OPS RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "test_*.py") string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}") -list(REMOVE_ITEM TEST_OPS test_conv2d_ngraph_op) - foreach(TEST_OP ${TEST_OPS}) py_test_modules(${TEST_OP} MODULES ${TEST_OP} ENVS FLAGS_use_ngraph=true) endforeach(TEST_OP) diff --git a/python/paddle/fluid/tests/unittests/ngraph/test_conv2d_ngraph_op.py b/python/paddle/fluid/tests/unittests/ngraph/test_conv2d_ngraph_op.py index 4894af949a20bad59836989060842c82b78d5cfe..7a9adcf4cd80ba7e1cbba1e9809530e6535a6ae7 100644 --- a/python/paddle/fluid/tests/unittests/ngraph/test_conv2d_ngraph_op.py +++ b/python/paddle/fluid/tests/unittests/ngraph/test_conv2d_ngraph_op.py @@ -20,6 +20,13 @@ from test_conv2d_op import TestConv2dOp, TestWithPad, TestWithStride, TestWithGr import numpy as np +class TestNGRAPHWithStride(TestWithStride): + def init_test_case(self): + super(TestNGRAPHWithStride, self).init_test_case() + self.use_cuda = False + self.dtype = np.float32 + + class TestNGRAPHDepthwiseConv(TestDepthwiseConv): def init_test_case(self): super(TestNGRAPHDepthwiseConv, self).init_test_case() @@ -55,7 +62,7 @@ class TestNGRAPHDepthwiseConvWithDilation2(TestDepthwiseConvWithDilation2): self.dtype = np.float32 -del TestDepthwiseConv, TestDepthwiseConv2, TestDepthwiseConv3, TestDepthwiseConvWithDilation, TestDepthwiseConvWithDilation2 +del TestWithStride, TestDepthwiseConv, TestDepthwiseConv2, TestDepthwiseConv3, TestDepthwiseConvWithDilation, TestDepthwiseConvWithDilation2 if __name__ == '__main__': unittest.main()