From aeec2a46fd1dc14b19e835911d7157c39dcdf205 Mon Sep 17 00:00:00 2001
From: Tom Shen <jun.shen@advance.ai>
Date: Sat, 24 Apr 2021 00:53:50 +0800
Subject: [PATCH] [BUGFIX] Fix Windows GPU VS2019 build (#20206)

---
 src/ndarray/ndarray_function-inl.h | 64 ++++++++++++++++++++++++++++--
 1 file changed, 60 insertions(+), 4 deletions(-)

diff --git a/src/ndarray/ndarray_function-inl.h b/src/ndarray/ndarray_function-inl.h
index 6ac586acd99c..d5bc62ec4771 100644
--- a/src/ndarray/ndarray_function-inl.h
+++ b/src/ndarray/ndarray_function-inl.h
@@ -230,8 +230,62 @@ void EvalRandom<DEVICE, GaussianDistribution>(
   }
 }
 
+#if defined(__CUDACC__)
+
 template<>
-void EvalRandom<DEVICE, GammaDistribution>(
+void EvalRandom<gpu, GammaDistribution>(
+    const real_t &alpha,
+    const real_t &beta,
+    const Resource &resource,
+    TBlob *ret,
+    RunContext ctx) {
+  EvalRandom<cpu, GammaDistribution>(alpha, beta, resource, ret, ctx);
+}
+
+template<>
+void EvalRandom<gpu, ExponentialDistribution>(
+    const real_t &lambda,
+    const real_t &dummy,  // this is to satisfy the SampleOp lambda signature
+    const Resource &resource,
+    TBlob *ret,
+    RunContext ctx) {
+  EvalRandom<cpu, ExponentialDistribution>(lambda, dummy, resource, ret, ctx);
+}
+
+template<>
+void EvalRandom<gpu, PoissonDistribution>(
+    const real_t &lambda,
+    const real_t &dummy,  // this is to satisfy the SampleOp lambda signature
+    const Resource &resource,
+    TBlob *ret,
+    RunContext ctx) {
+  EvalRandom<cpu, PoissonDistribution>(lambda, dummy, resource, ret, ctx);
+}
+
+template<>
+void EvalRandom<gpu, NegBinomialDistribution>(
+    const real_t &k,
+    const real_t &p,
+    const Resource &resource,
+    TBlob *ret,
+    RunContext ctx) {
+  EvalRandom<cpu, NegBinomialDistribution>(k, p, resource, ret, ctx);
+}
+
+template<>
+void EvalRandom<gpu, GenNegBinomialDistribution>(
+    const real_t &mu,
+    const real_t &alpha,
+    const Resource &resource,
+    TBlob *ret,
+    RunContext ctx) {
+  EvalRandom<cpu, GenNegBinomialDistribution>(mu, alpha, resource, ret, ctx);
+}
+
+#else
+
+template<>
+void EvalRandom<cpu, GammaDistribution>(
     const real_t &alpha,
     const real_t &beta,
     const Resource &resource,
@@ -261,7 +315,7 @@ void EvalRandom<DEVICE, GammaDistribution>(
 
 
 template<>
-void EvalRandom<DEVICE, ExponentialDistribution>(
+void EvalRandom<cpu, ExponentialDistribution>(
     const real_t &lambda,
     const real_t &dummy,  // this is to satisfy the SampleOp lambda signature
     const Resource &resource,
@@ -290,7 +344,7 @@ void EvalRandom<DEVICE, ExponentialDistribution>(
 }
 
 template<>
-void EvalRandom<DEVICE, PoissonDistribution>(
+void EvalRandom<cpu, PoissonDistribution>(
     const real_t &lambda,
     const real_t &dummy,  // this is to satisfy the SampleOp lambda signature
     const Resource &resource,
@@ -319,7 +373,7 @@ void EvalRandom<DEVICE, PoissonDistribution>(
 }
 
 template<>
-void EvalRandom<DEVICE, NegBinomialDistribution>(
+void EvalRandom<cpu, NegBinomialDistribution>(
     const real_t &k,
     const real_t &p,
     const Resource &resource,
@@ -376,6 +430,8 @@ void EvalRandom<DEVICE, GenNegBinomialDistribution>(
   }
 }
 
+#endif // #ifndef __CUDACC__
+
 template<>
 void Eval<DEVICE>(const real_t &rhs, TBlob *ret, RunContext ctx) {
   mshadow::Stream<DEVICE> *s = ctx.get_stream<DEVICE>();