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

Commit

Permalink
fix bugs
Browse files Browse the repository at this point in the history
  • Loading branch information
szha committed Jul 25, 2020
1 parent 36ba5c5 commit 74a7b9e
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 5 deletions.
3 changes: 2 additions & 1 deletion src/api/operator/numpy/np_matrix_op.cc
Original file line number Diff line number Diff line change
Expand Up @@ -613,6 +613,7 @@ MXNET_REGISTER_API("_npi.tril_indices")
ndarray_handles.emplace_back(ndoutputs[i]);
}
*ret = ADT(0, ndarray_handles.begin(), ndarray_handles.end());
});

MXNET_REGISTER_API("_npi.vstack")
.set_body([](runtime::MXNetArgs args, runtime::MXNetRetValue* ret) {
Expand All @@ -627,7 +628,7 @@ MXNET_REGISTER_API("_npi.vstack")
SetAttrDict<op::NumpyVstackParam>(&attrs);
int num_outputs = 0;
std::vector<NDArray*> inputs_vec(args.size(), nullptr);
for (int i = 0; i < args_size; ++i) {
for (int i = 0; i < args.size(); ++i) {
inputs_vec[i] = args[i].operator mxnet::NDArray*();
}
NDArray** inputs = inputs_vec.data();
Expand Down
6 changes: 3 additions & 3 deletions src/api/operator/numpy/random/np_multinomial_op.cc
Original file line number Diff line number Diff line change
Expand Up @@ -24,8 +24,8 @@
#include <mxnet/api_registry.h>
#include <mxnet/runtime/packed_func.h>
#include <vector>
#include "../utils.h"
#include "../../../operator/numpy/random/np_multinomial_op.h"
#include "../../utils.h"
#include "../../../../operator/numpy/random/np_multinomial_op.h"

namespace mxnet {

Expand All @@ -39,7 +39,7 @@ MXNET_REGISTER_API("_npi.multinomial")
int num_inputs = 0;

// parse int
param.n = arg[0].operator int();
param.n = args[0].operator int();

// parse pvals
if (args[1].type_code() == kNull) {
Expand Down
2 changes: 1 addition & 1 deletion src/operator/numpy/random/np_multinomial_op.h
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ struct NumpyMultinomialParam : public dmlc::Parameter<NumpyMultinomialParam> {
"Default is None, in which case a single value is returned.");
}
void SetAttrDict(std::unordered_map<std::string, std::string>* dict) {
std::ostringstream n, pvals, size;
std::ostringstream n_s, pvals_s, size_s;
n_s << n;
pvals_s << pvals;
size_s << size;
Expand Down

0 comments on commit 74a7b9e

Please sign in to comment.