From 44458865b686f5b8fc67b02be22825df508990e6 Mon Sep 17 00:00:00 2001 From: gongchen Date: Wed, 20 May 2020 11:03:18 +0800 Subject: [PATCH] test(custom_op): Clean up custom op test case. --- tests/st/ops/custom_ops_tbe/__init__.py | 0 .../ops/custom_ops_tbe/{cus_conv2d_impl.py => conv2d_impl.py} | 0 tests/st/ops/custom_ops_tbe/cus_conv2d.py | 2 +- tests/st/ops/custom_ops_tbe/{test_cus_conv.py => test_conv.py} | 0 4 files changed, 1 insertion(+), 1 deletion(-) delete mode 100644 tests/st/ops/custom_ops_tbe/__init__.py rename tests/st/ops/custom_ops_tbe/{cus_conv2d_impl.py => conv2d_impl.py} (100%) rename tests/st/ops/custom_ops_tbe/{test_cus_conv.py => test_conv.py} (100%) diff --git a/tests/st/ops/custom_ops_tbe/__init__.py b/tests/st/ops/custom_ops_tbe/__init__.py deleted file mode 100644 index e69de29bb2d..00000000000 diff --git a/tests/st/ops/custom_ops_tbe/cus_conv2d_impl.py b/tests/st/ops/custom_ops_tbe/conv2d_impl.py similarity index 100% rename from tests/st/ops/custom_ops_tbe/cus_conv2d_impl.py rename to tests/st/ops/custom_ops_tbe/conv2d_impl.py diff --git a/tests/st/ops/custom_ops_tbe/cus_conv2d.py b/tests/st/ops/custom_ops_tbe/cus_conv2d.py index 8cf8177ecba..0002ae3af12 100644 --- a/tests/st/ops/custom_ops_tbe/cus_conv2d.py +++ b/tests/st/ops/custom_ops_tbe/cus_conv2d.py @@ -103,7 +103,7 @@ class Cus_Conv2D(PrimitiveWithInfer): raise ValueError('Attr \'kernel_size\' of \'Conv2D\' Op passed ' + str(self.kernel_size) + ', should be a int or tuple and equal to or greater than 1.') self.stride = validator.check_integer('stride', stride, 1, Rel.GE) - from .cus_conv2d_impl import Cus_Conv2D + from conv2d_impl import Cus_Conv2D def infer_shape(self, x_shape, w_shape): validator.check_integer("weight_shape", len(w_shape), 4, Rel.EQ) diff --git a/tests/st/ops/custom_ops_tbe/test_cus_conv.py b/tests/st/ops/custom_ops_tbe/test_conv.py similarity index 100% rename from tests/st/ops/custom_ops_tbe/test_cus_conv.py rename to tests/st/ops/custom_ops_tbe/test_conv.py