sample_uniform revert

apache · Nov 15, 2018 · 9272bf6 · 9272bf6
1 parent e7e622c
commit 9272bf6
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/include/mxnet/random_generator.h b/include/mxnet/random_generator.h
@@ -55,8 +55,6 @@ class RandGenerator<cpu, DType> {
    public:
     typedef typename std::conditional<std::is_floating_point<DType>::value,
                                       DType, double>::type FType;
-    typedef typename std::conditional<std::is_integral<DType>::value,
-                                      DType, int>::type IType;
     explicit Impl(RandGenerator<cpu, DType> *gen, int state_idx)
         : engine_(gen->states_ + state_idx) {}
 

diff --git a/python/mxnet/ndarray/random.py b/python/mxnet/ndarray/random.py
@@ -565,5 +565,5 @@ def randint(low, high, shape=_Null, dtype=_Null, ctx=None, out=None, **kwargs):
      [ 3  3]]
     <NDArray 3x2 @cpu(0)>
     """
-    return _random_helper(_internal._random_randint, None,
+    return _random_helper(_internal._random_randint, _internal._sample_uniform,
                           [low, high], shape, dtype, ctx, out, kwargs)