From 27ceb43bf6b06dda898e5d027097f33a970f7355 Mon Sep 17 00:00:00 2001
From: Benoit Steiner <benoit.steiner.goog@gmail.com>
Date: Mon, 19 Dec 2016 15:34:42 -0800
Subject: Fixed race condition in the tensor_shuffling_sycl test

---
 unsupported/test/cxx11_tensor_shuffling_sycl.cpp | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'unsupported/test/cxx11_tensor_shuffling_sycl.cpp')

diff --git a/unsupported/test/cxx11_tensor_shuffling_sycl.cpp b/unsupported/test/cxx11_tensor_shuffling_sycl.cpp
index b2b75cbde..c4521aac8 100644
--- a/unsupported/test/cxx11_tensor_shuffling_sycl.cpp
+++ b/unsupported/test/cxx11_tensor_shuffling_sycl.cpp
@@ -57,6 +57,7 @@ static void test_simple_shuffling_sycl(const Eigen::SyclDevice& sycl_device)
 
   gpu2.device(sycl_device)=gpu1.shuffle(shuffles);
   sycl_device.memcpyDeviceToHost(no_shuffle.data(), gpu_data2, buffSize);
+  sycl_device.synchronize();
 
   VERIFY_IS_EQUAL(no_shuffle.dimension(0), sizeDim1);
   VERIFY_IS_EQUAL(no_shuffle.dimension(1), sizeDim2);
@@ -82,8 +83,9 @@ static void test_simple_shuffling_sycl(const Eigen::SyclDevice& sycl_device)
   DataType* gpu_data3  = static_cast<DataType*>(sycl_device.allocate(buffSize));
   TensorMap<Tensor<DataType, 4,DataLayout,IndexTypes>> gpu3(gpu_data3, tensorrangeShuffle);
 
-    gpu3.device(sycl_device)=gpu1.shuffle(shuffles);
-    sycl_device.memcpyDeviceToHost(shuffle.data(), gpu_data3, buffSize);
+  gpu3.device(sycl_device)=gpu1.shuffle(shuffles);
+  sycl_device.memcpyDeviceToHost(shuffle.data(), gpu_data3, buffSize);
+  sycl_device.synchronize();
 
   VERIFY_IS_EQUAL(shuffle.dimension(0), sizeDim3);
   VERIFY_IS_EQUAL(shuffle.dimension(1), sizeDim4);
-- 
cgit v1.2.3