update test (apache#48)

tqchen · May 26, 2018 · 7660fd1 · 7660fd1
1 parent 47cacbf
commit 7660fd1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nnvm/tests/cpp/op_test.cc b/nnvm/tests/cpp/op_test.cc
@@ -6,10 +6,10 @@
 NNVM_REGISTER_OP(add)
 .describe("add two data together")
 .set_num_inputs(2)
-.attr("inplace_pair", std::make_pair(0, 0));
+.set_attr("inplace_pair", std::make_pair(0, 0));
 
 NNVM_REGISTER_OP(add)
-.attr<std::string>("nick_name", "plus");
+.set_attr<std::string>("nick_name", "plus");
 
 
 TEST(Op, GetAttr) {