From 71a604fae3970ff0422f45617f57d85027cdddce Mon Sep 17 00:00:00 2001
From: Rory Mitchell <r.a.mitchell.nz@gmail.com>
Date: Sun, 17 Feb 2019 19:42:32 +1300
Subject: [PATCH] Fix for windows compilation (#4139)

---
 src/common/device_helpers.cuh | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/common/device_helpers.cuh b/src/common/device_helpers.cuh
index 62f0a2229..1d6355c54 100644
--- a/src/common/device_helpers.cuh
+++ b/src/common/device_helpers.cuh
@@ -1050,8 +1050,9 @@ class SaveCudaContext {
 template <typename T, typename FunctionT>
 void ExecuteIndexShards(std::vector<T> *shards, FunctionT f) {
   SaveCudaContext{[&]() {
-#pragma omp parallel for schedule(static, 1) if (shards->size() > 1)
-    for (size_t shard = 0; shard < shards->size(); ++shard) {
+    const long shards_size = static_cast<long>(shards->size());
+#pragma omp parallel for schedule(static, 1) if (shards_size > 1)
+    for (long shard = 0; shard < shards_size; ++shard) {
       f(shard, shards->at(shard));
     }
   }};