From 556f83dedddf7cd0df6892dd9bfa0ad0ed85c850 Mon Sep 17 00:00:00 2001
From: Alexander Zai <alexzai@amazon.com>
Date: Tue, 28 Aug 2018 02:00:39 -0700
Subject: [PATCH] fix template

---
 include/mxnet/random_generator.h | 2 +-
 src/common/random_generator.cu   | 8 ++++----
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/include/mxnet/random_generator.h b/include/mxnet/random_generator.h
index 49c1b2dddd49..db1daf638d1b 100644
--- 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
index ec7bedb120f4..b3a225184d22 100644
--- 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_));
 }