fix row_conv_op small size unittest cases (#22606)

revert-22710-feature/integrated_ps_api
Double_V 5 years ago committed by GitHub
parent e136661304
commit 163a3ddfe3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -50,7 +50,7 @@ class TestRowConvOp1(OpTest):
lod = [[2, 3, 2]]
T = sum(lod[0])
D = 16
context_length = 2
context_length = 8
x = np.random.random((T, D)).astype("float32")
wt = np.random.random((context_length, D)).astype("float32")
@ -139,11 +139,11 @@ def row_conv_foward_Tensor(x, wt):
class TestRowOpWithTensorInput(OpTest):
def setUp(self):
self.op_type = "row_conv"
length = [3, 2, 4]
length = [1, 2, 3]
B = 2
T = sum(length)
D = 16
context_length = 2
D = 20
context_length = 6
x = np.random.random((B, T, D)).astype("float32")
wt = np.random.random((context_length, D)).astype("float32")

@ -25,7 +25,6 @@ NEED_TO_FIX_OP_LIST = [
'margin_rank_loss',
'matmul',
'mul',
'row_conv',
'scatter',
'smooth_l1_loss',
'soft_relu',

Loading…
Cancel
Save