From 56e2b0cd4e9748963afcbb1e55bdd2438310c7d0 Mon Sep 17 00:00:00 2001
From: Aleksei Iupinov <a.yupinov@gmail.com>
Date: Tue, 21 Nov 2017 08:19:26 +0100
Subject: [PATCH] Fix pinned memory status checker tests

Change-Id: I69df298c5c022a21fe7e96d201adfe1d4dfb8563
---
 src/gromacs/gpu_utils/tests/pinnedmemorychecker.cpp | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/gromacs/gpu_utils/tests/pinnedmemorychecker.cpp b/src/gromacs/gpu_utils/tests/pinnedmemorychecker.cpp
index 1816b3337e..0f34dab15f 100644
--- a/src/gromacs/gpu_utils/tests/pinnedmemorychecker.cpp
+++ b/src/gromacs/gpu_utils/tests/pinnedmemorychecker.cpp
@@ -72,15 +72,15 @@ TEST_F(PinnedMemoryCheckerTest, DefaultContainerIsRecognized)
 
 TEST_F(PinnedMemoryCheckerTest, NonpinnedContainerIsRecognized)
 {
-    auto allocator = Allocator<real, HostAllocationPolicy>(HostAllocationPolicy());
-    std::vector<real, decltype(allocator)> dummy(3, 1.5, allocator);
+    HostVector<real> dummy(3, 1.5);
+    changePinningPolicy(&dummy, PinningPolicy::CannotBePinned);
     EXPECT_FALSE(isHostMemoryPinned(dummy.data()));
 }
 
 TEST_F(PinnedMemoryCheckerTest, PinnedContainerIsRecognized)
 {
-    auto allocator = Allocator<real, HostAllocationPolicy>(makeHostAllocationPolicyForGpu());
-    std::vector<real, decltype(allocator)> dummy(3, 1.5, allocator);
+    HostVector<real> dummy(3, 1.5);
+    changePinningPolicy(&dummy, PinningPolicy::CanBePinned);
     EXPECT_TRUE(isHostMemoryPinned(dummy.data()));
 }
 
-- 
2.11.4.GIT