fix template

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

Hide file tree

Showing 2 changed files with 7 additions and 7 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];
   }
 
@@ -149,9 +149,9 @@ class RandGenerator<gpu, DType> {
     curandStatePhilox4_32_10_t state_;
   };  // class RandGenerator<gpu, DType>::Impl
 
-  static void AllocState(RandGenerator<gpu, DType> *inst) {}
+  static void AllocState(RandGenerator<gpu, DType> *inst);
 
-  static void FreeState(RandGenerator<gpu, DType> *inst) {}
+  static void FreeState(RandGenerator<gpu, DType> *inst);
 
   void Seed(mshadow::Stream<gpu> *s, uint32_t seed);
 

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<>
-void RandGenerator<gpu, float>::AllocState(RandGenerator<gpu, float> *inst) {
+template<typename DType>
+void RandGenerator<gpu, DType>::AllocState(RandGenerator<gpu, DType> *inst) {
   CUDA_CALL(cudaMalloc(&inst->states_,
                        kNumRandomStates * sizeof(curandStatePhilox4_32_10_t)));
 }
 
-template<>
-void RandGenerator<gpu, float>::FreeState(RandGenerator<gpu, float> *inst) {
+template<typename DType>
+void RandGenerator<gpu, DType>::FreeState(RandGenerator<gpu, DType> *inst) {
   CUDA_CALL(cudaFree(inst->states_));
 }