Skip to content

Commit 50dab46

Browse files
committed
Fix bug
1 parent 23e19e2 commit 50dab46

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

python/paddle/fluid/tests/unittests/test_operator_desc.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,10 @@ def test_op_desc_creation(self):
6363
self.assertEqual(mul_op.output("Out"), ["mul.out"])
6464
self.assertEqual(
6565
set(mul_op.attr_names),
66-
set(["x_num_col_dims", "y_num_col_dims", "use_mkldnn"]))
66+
set([
67+
"x_num_col_dims", "y_num_col_dims", "use_mkldnn", "op_role",
68+
"op_role_var"
69+
]))
6770
self.assertEqual(mul_op.has_attr("x_num_col_dims"), True)
6871
self.assertEqual(mul_op.attr_type("x_num_col_dims"), core.AttrType.INT)
6972
self.assertEqual(mul_op.attr("x_num_col_dims"), 1)

0 commit comments

Comments
 (0)