fix template arg issue windows fix int32 default

apache · Oct 15, 2018 · 12f92a3 · 12f92a3
1 parent 2e24870
commit 12f92a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/include/mxnet/random_generator.h b/include/mxnet/random_generator.h
@@ -74,7 +74,7 @@ class RandGenerator<cpu, DType> {
     MSHADOW_XINLINE DType discrete_uniform(const int lower, const int upper) {
       typedef typename std::conditional<std::is_integral<DType>::value,
       std::uniform_int_distribution<DType>,
-      std::uniform_real_distribution<FType>>::type GType;
+      std::uniform_int_distribution<kInt32>>::type GType;
       GType dist_discrete_uniform(lower, upper);
       return dist_discrete_uniform(*engine_);
     }