add occupied(n) for unordered set and map

stotko · Aug 8, 2024 · 87fe7d7 · 87fe7d7
1 parent 3b7d712
commit 87fe7d7
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 0 deletions.
diff --git a/src/stdgpu/impl/unordered_map_detail.cuh b/src/stdgpu/impl/unordered_map_detail.cuh
@@ -358,6 +358,13 @@ unordered_map<Key, T, Hash, KeyEqual, Allocator>::clear(ExecutionPolicy&& policy
     _base.clear(std::forward<ExecutionPolicy>(policy));
 }
 
+template <typename Key, typename T, typename Hash, typename KeyEqual, typename Allocator>
+inline STDGPU_DEVICE_ONLY bool
+unordered_map<Key, T, Hash, KeyEqual, Allocator>::occupied(const index_t n) const
+{
+    return _base.occupied(n);
+}
+
 template <typename Key, typename T, typename Hash, typename KeyEqual, typename Allocator>
 unordered_map<Key, T, Hash, KeyEqual, Allocator>
 unordered_map<Key, T, Hash, KeyEqual, Allocator>::createDeviceObject(const index_t& capacity,

diff --git a/src/stdgpu/impl/unordered_set_detail.cuh b/src/stdgpu/impl/unordered_set_detail.cuh
@@ -340,6 +340,13 @@ unordered_set<Key, Hash, KeyEqual, Allocator>::clear(ExecutionPolicy&& policy)
     _base.clear(std::forward<ExecutionPolicy>(policy));
 }
 
+template <typename Key, typename Hash, typename KeyEqual, typename Allocator>
+inline STDGPU_DEVICE_ONLY bool
+unordered_set<Key, Hash, KeyEqual, Allocator>::occupied(const index_t n) const
+{
+    return _base.occupied(n);
+}
+
 template <typename Key, typename Hash, typename KeyEqual, typename Allocator>
 unordered_set<Key, Hash, KeyEqual, Allocator>
 unordered_set<Key, Hash, KeyEqual, Allocator>::createDeviceObject(const index_t& capacity, const Allocator& allocator)

diff --git a/src/stdgpu/unordered_map.cuh b/src/stdgpu/unordered_map.cuh
@@ -484,6 +484,14 @@ public:
     STDGPU_HOST_DEVICE key_equal
     key_eq() const;
 
+    /**
+     * \brief Checks if the given bucket is occupied
+     * \param[in] n The bucket index
+     * \return True if the bucket is occupied, false otherwise
+     */
+    STDGPU_DEVICE_ONLY bool
+    occupied(const index_t n) const;
+
 private:
     using base_type =
             detail::unordered_base<key_type, value_type, detail::select1st<value_type>, hasher, key_equal, Allocator>;

diff --git a/src/stdgpu/unordered_set.cuh b/src/stdgpu/unordered_set.cuh
@@ -473,6 +473,14 @@ public:
     STDGPU_HOST_DEVICE key_equal
     key_eq() const;
 
+    /**
+     * \brief Checks if the given bucket is occupied
+     * \param[in] n The bucket index
+     * \return True if the bucket is occupied, false otherwise
+     */
+    STDGPU_DEVICE_ONLY bool
+    occupied(const index_t n) const;
+
 private:
     using base_type = detail::unordered_base<key_type, value_type, identity, hasher, key_equal, Allocator>;