misc fix

example/extensions/lib_custom_op/gemm_lib.cc

@@ -71,6 +71,9 @@ MXReturnValue forward(std::map<std::string, std::string> attrs,
  unsigned k = inputs[0].shape[1];
  unsigned m = inputs[1].shape[1];
 
+ int random = res.get_randint();
+ std::cout << random << std::endl;
+
  gemm(A, B, C, n, k, m);
  }
  return MX_SUCCESS;

example/extensions/lib_custom_op/relu_lib.cu

@@ -44,8 +44,6 @@ MXReturnValue forwardCPU(std::map<std::string, std::string> attrs,
  OpResource res) {
  float* in_data = inputs[0].data<float>();
  float* out_data = outputs[0].data<float>();
- int random = res.get_randint();
- std::cout << random << std::endl;
  for (int i=0; i<inputs[0].size(); i++) {
  out_data[i] = in_data[i] > 0 ? in_data[i] : 0;
  }

include/mxnet/lib_api.h

@@ -369,7 +369,7 @@ union RandomType {
  double d;
 };
 
-typedef RandomType (*rng_caller_t)(void*, char*);
+typedef RandomType (*rng_caller_t)(void*, const char*);
 
 #if defined(__NVCC__)
  typedef cudaStream_t mx_stream_t;
@@ -404,7 +404,7 @@ class OpResource {
  return static_cast<mx_stream_t>(cuda_stream);
  }
 
- int gen_randint() {
+ int get_randint() {
  RandomType ret = rng_caller_nocap(rng_caller, "rand");
  return ret.i;
  }

src/c_api/c_api.cc

@@ -181,7 +181,7 @@ void CustomFComputeDispatcher(const std::string op_name,
  mxnet::common::random::RandGenerator<cpu, double> *pgen =
  ctx.requested[1].get_parallel_random<cpu, double>();
 
- auto rng_caller = [&](char *rand_type) {
+ auto rng_caller = [&](const char *rand_type) {
  LOG(INFO) << "rng_caller called";
  typename mxnet::common::random::RandGenerator<cpu, double>::Impl genImpl(pgen, 1);
  std::string rand_str(rand_type);
@@ -194,7 +194,7 @@ void CustomFComputeDispatcher(const std::string op_name,
  };
 
  typedef decltype(rng_caller) type_rng_caller;
- auto rng_caller_nocap = [](void *rng_call, char *rand_type) {
+ auto rng_caller_nocap = [](void *rng_call, const char *rand_type) {
  type_rng_caller* rngcaller = static_cast<type_rng_caller*>(rng_call);
  return (*rngcaller)(rand_type);
  };