未验证 提交 163a3ddf 编写于 作者: D Double_V 提交者: GitHub

fix row_conv_op small size unittest cases (#22606)

上级 e1366613
...@@ -50,7 +50,7 @@ class TestRowConvOp1(OpTest): ...@@ -50,7 +50,7 @@ class TestRowConvOp1(OpTest):
lod = [[2, 3, 2]] lod = [[2, 3, 2]]
T = sum(lod[0]) T = sum(lod[0])
D = 16 D = 16
context_length = 2 context_length = 8
x = np.random.random((T, D)).astype("float32") x = np.random.random((T, D)).astype("float32")
wt = np.random.random((context_length, D)).astype("float32") wt = np.random.random((context_length, D)).astype("float32")
...@@ -139,11 +139,11 @@ def row_conv_foward_Tensor(x, wt): ...@@ -139,11 +139,11 @@ def row_conv_foward_Tensor(x, wt):
class TestRowOpWithTensorInput(OpTest): class TestRowOpWithTensorInput(OpTest):
def setUp(self): def setUp(self):
self.op_type = "row_conv" self.op_type = "row_conv"
length = [3, 2, 4] length = [1, 2, 3]
B = 2 B = 2
T = sum(length) T = sum(length)
D = 16 D = 20
context_length = 2 context_length = 6
x = np.random.random((B, T, D)).astype("float32") x = np.random.random((B, T, D)).astype("float32")
wt = np.random.random((context_length, D)).astype("float32") wt = np.random.random((context_length, D)).astype("float32")
......
...@@ -25,7 +25,6 @@ NEED_TO_FIX_OP_LIST = [ ...@@ -25,7 +25,6 @@ NEED_TO_FIX_OP_LIST = [
'margin_rank_loss', 'margin_rank_loss',
'matmul', 'matmul',
'mul', 'mul',
'row_conv',
'scatter', 'scatter',
'smooth_l1_loss', 'smooth_l1_loss',
'soft_relu', 'soft_relu',
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册