提交 1c39efb7 编写于 作者: L Leo Zhao 提交者: Tao Luo

Enable test conv2d ngraph (#22074)

上级 c112b645
file(GLOB TEST_OPS RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "test_*.py") file(GLOB TEST_OPS RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "test_*.py")
string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}") string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")
list(REMOVE_ITEM TEST_OPS test_conv2d_ngraph_op)
foreach(TEST_OP ${TEST_OPS}) foreach(TEST_OP ${TEST_OPS})
py_test_modules(${TEST_OP} MODULES ${TEST_OP} ENVS FLAGS_use_ngraph=true) py_test_modules(${TEST_OP} MODULES ${TEST_OP} ENVS FLAGS_use_ngraph=true)
endforeach(TEST_OP) endforeach(TEST_OP)
...@@ -20,6 +20,13 @@ from test_conv2d_op import TestConv2dOp, TestWithPad, TestWithStride, TestWithGr ...@@ -20,6 +20,13 @@ from test_conv2d_op import TestConv2dOp, TestWithPad, TestWithStride, TestWithGr
import numpy as np 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): class TestNGRAPHDepthwiseConv(TestDepthwiseConv):
def init_test_case(self): def init_test_case(self):
super(TestNGRAPHDepthwiseConv, self).init_test_case() super(TestNGRAPHDepthwiseConv, self).init_test_case()
...@@ -55,7 +62,7 @@ class TestNGRAPHDepthwiseConvWithDilation2(TestDepthwiseConvWithDilation2): ...@@ -55,7 +62,7 @@ class TestNGRAPHDepthwiseConvWithDilation2(TestDepthwiseConvWithDilation2):
self.dtype = np.float32 self.dtype = np.float32
del TestDepthwiseConv, TestDepthwiseConv2, TestDepthwiseConv3, TestDepthwiseConvWithDilation, TestDepthwiseConvWithDilation2 del TestWithStride, TestDepthwiseConv, TestDepthwiseConv2, TestDepthwiseConv3, TestDepthwiseConvWithDilation, TestDepthwiseConvWithDilation2
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册