From 3a63c2139f66b63a6d559f33655f792e9ee6e729 Mon Sep 17 00:00:00 2001
From: Tom Shen <sjiagc@gmail.com>
Date: Fri, 30 Apr 2021 22:31:18 +0800
Subject: [PATCH] [BUGFIX] Fix Windows GPU VS2019 build (#20206) (#20207)

Co-authored-by: Tom Shen <jun.shen@advance.ai>
---
 src/ndarray/ndarray_function-inl.h | 67 +++++++++++++++++++++++++++---
 1 file changed, 61 insertions(+), 6 deletions(-)

diff --git a/src/ndarray/ndarray_function-inl.h b/src/ndarray/ndarray_function-inl.h
index 6ac586acd99c..beb40c73b8ad 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<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<DEVICE, GammaDistribution>(
+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,
@@ -259,9 +313,8 @@ 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 +343,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 +372,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,
@@ -348,7 +401,7 @@ void EvalRandom<DEVICE, NegBinomialDistribution>(
 }
 
 template<>
-void EvalRandom<DEVICE, GenNegBinomialDistribution>(
+void EvalRandom<cpu, GenNegBinomialDistribution>(
     const real_t &mu,
     const real_t &alpha,
     const Resource &resource,
@@ -376,6 +429,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>();