From 8341db44803fefb053fd2b35e4125e2d4a6be414 Mon Sep 17 00:00:00 2001
From: Jake Hemstad <jhemstad@nvidia.com>
Date: Tue, 2 Mar 2021 18:40:45 -0600
Subject: [PATCH] Remove dependence on managed memory for multimap test (#7451)

Removes managed memory from the `HASH_MAP_TEST` to allow the test to work on systems that don't support managed memory.

Authors:
  - Jake Hemstad (@jrhemstad)

Approvers:
  - Ram (Ramakrishna Prabhu) (@rgsl888prabhu)
  - Mark Harris (@harrism)

URL: https://github.com/rapidsai/cudf/pull/7451
---
 cpp/tests/hash_map/multimap_test.cu | 24 ++++++++++--------------
 1 file changed, 10 insertions(+), 14 deletions(-)

diff --git a/cpp/tests/hash_map/multimap_test.cu b/cpp/tests/hash_map/multimap_test.cu
index 1f1d20eb8ae..7fd69e90efd 100644
--- a/cpp/tests/hash_map/multimap_test.cu
+++ b/cpp/tests/hash_map/multimap_test.cu
@@ -15,6 +15,7 @@
  */
 
 #include <hash/concurrent_unordered_multimap.cuh>
+#include <hash/hash_allocator.cuh>
 
 #include <cudf_test/base_fixture.hpp>
 
@@ -46,11 +47,15 @@ class MultimapTest : public cudf::test::BaseFixture {
   using value_type = typename T::value_type;
   using size_type  = int;
 
-  using multimap_type = concurrent_unordered_multimap<key_type,
-                                                      value_type,
-                                                      size_type,
-                                                      std::numeric_limits<key_type>::max(),
-                                                      std::numeric_limits<value_type>::max()>;
+  using multimap_type =
+    concurrent_unordered_multimap<key_type,
+                                  value_type,
+                                  size_type,
+                                  std::numeric_limits<key_type>::max(),
+                                  std::numeric_limits<value_type>::max(),
+                                  default_hash<key_type>,
+                                  equal_to<key_type>,
+                                  default_allocator<thrust::pair<key_type, value_type>>>;
 
   std::unique_ptr<multimap_type, std::function<void(multimap_type*)>> the_map;
 
@@ -91,12 +96,3 @@ TYPED_TEST(MultimapTest, InitialState)
   auto end   = this->the_map->end();
   EXPECT_NE(begin, end);
 }
-
-TYPED_TEST(MultimapTest, CheckUnusedValues)
-{
-  EXPECT_EQ(this->the_map->get_unused_key(), this->unused_key);
-
-  auto begin = this->the_map->begin();
-  EXPECT_EQ(begin->first, this->unused_key);
-  EXPECT_EQ(begin->second, this->unused_value);
-}