Commit 154cb98c by Tianqi Chen

update test (#48)

parent c30ef503
......@@ -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) {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment