未验证 提交 18de0c94 编写于 作者: H HongyuJia 提交者: GitHub

[0D-Tensor] Fix test_elementwise_max_op unittest of FP16 (#55683)

上级 aa1f4a44
......@@ -130,9 +130,6 @@ class TestElementwiseMaxFP16Op_ZeroDim1(TestElementwiseFP16Op):
self.x = np.random.uniform(0.1, 1, []).astype(np.float16)
self.y = np.random.uniform(0.1, 1, []).astype(np.float16)
def if_enbale_cinn(self):
self.enable_cinn = False
class TestElementwiseMaxOp_ZeroDim2(TestElementwiseOp):
def init_data(self):
......@@ -145,9 +142,6 @@ class TestElementwiseMaxFP16Op_ZeroDim2(TestElementwiseFP16Op):
self.x = np.random.uniform(0.1, 1, [13, 17]).astype(np.float16)
self.y = np.random.uniform(0.1, 1, []).astype(np.float16)
def if_enbale_cinn(self):
self.enable_cinn = False
class TestElementwiseMaxOp_ZeroDim3(TestElementwiseOp):
def init_data(self):
......@@ -160,9 +154,6 @@ class TestElementwiseMaxFP16Op_ZeroDim3(TestElementwiseFP16Op):
self.x = np.random.uniform(0.1, 1, []).astype(np.float16)
self.y = np.random.uniform(0.1, 1, [13, 17]).astype(np.float16)
def if_enbale_cinn(self):
self.enable_cinn = False
@unittest.skipIf(
core.is_compiled_with_cuda()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册