Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
* Fix clang
Browse files Browse the repository at this point in the history
  • Loading branch information
hanke580 committed Apr 20, 2020
1 parent 3ae5387 commit 7dd1b8f
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/api/operator/numpy/random/np_multinomial_op.cc
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,8 @@ MXNET_REGISTER_API("_npi.multinomial")
attrs.parsed = std::move(param);
attrs.op = op;
SetAttrDict<op::NumpyMultinomialParam>(&attrs);
auto ndoutputs = Invoke(op, &attrs, 0, nullptr, 0, nullptr);
int num_outputs = 0;
auto ndoutputs = Invoke(op, &attrs, 0, nullptr, &num_outputs, nullptr);
*ret = ndoutputs[0];
});

Expand Down

0 comments on commit 7dd1b8f

Please sign in to comment.