fix template

apache · Aug 28, 2018 · 556f83d · 556f83d
1 parent 786af1b
commit 556f83d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/include/mxnet/random_generator.h b/include/mxnet/random_generator.h
@@ -79,7 +79,7 @@ class RandGenerator<cpu, DType> {
     std::mt19937 *engine_;
   };
 
-  static void  AllocState(RandGenerator<cpu, DType> *inst) {
+  static void AllocState(RandGenerator<cpu, DType> *inst) {
     inst->states_ = new std::mt19937[kNumRandomStates];
   }
 

diff --git a/src/common/random_generator.cu b/src/common/random_generator.cu
@@ -59,14 +59,14 @@ void RandGenerator<gpu, float>::Seed(mshadow::Stream<gpu> *s, uint32_t seed) {
   s->Wait();
 }
 
-template<typename DType>
-void RandGenerator<gpu, DType>::AllocState(RandGenerator<gpu, DType> *inst) {
+template<typename>
+void RandGenerator<gpu, float>::AllocState(RandGenerator<gpu> *inst) {
   CUDA_CALL(cudaMalloc(&inst->states_,
                        kNumRandomStates * sizeof(curandStatePhilox4_32_10_t)));
 }
 
-template<typename DType>
-void RandGenerator<gpu, DType>::FreeState(RandGenerator<gpu, DType> *inst) {
+template<>
+void RandGenerator<gpu, float>::FreeState(RandGenerator<gpu> *inst) {
   CUDA_CALL(cudaFree(inst->states_));
 }