[dpdk-dev] [RFC 1/2] hash: add lock free support for extendable bucket
Wang, Yipeng1
yipeng1.wang at intel.com
Thu Mar 7 18:49:34 CET 2019
Thanks for this patch! Some initial comments inlined:
>-----Original Message-----
>From: Dharmik Thakkar [mailto:dharmik.thakkar at arm.com]
>Sent: Friday, March 1, 2019 4:24 PM
>To: Wang, Yipeng1 <yipeng1.wang at intel.com>; Gobriel, Sameh <sameh.gobriel at intel.com>; Richardson, Bruce
><bruce.richardson at intel.com>; De Lara Guarch, Pablo <pablo.de.lara.guarch at intel.com>; Mcnamara, John
><john.mcnamara at intel.com>; Kovacevic, Marko <marko.kovacevic at intel.com>
>Cc: dev at dpdk.org; Dharmik Thakkar <dharmik.thakkar at arm.com>
>Subject: [RFC 1/2] hash: add lock free support for extendable bucket
>
>This patch enables lock-free read-write concurrency support for
>extendable bucket feature.
>
>Suggested-by: Honnappa Nagarahalli <honnappa.nagarahalli at arm.com>
>Signed-off-by: Dharmik Thakkar <dharmik.thakkar at arm.com>
>---
> doc/guides/prog_guide/hash_lib.rst | 3 +-
> lib/librte_hash/rte_cuckoo_hash.c | 150 +++++++++++++++++++----------
> lib/librte_hash/rte_cuckoo_hash.h | 8 ++
> 3 files changed, 110 insertions(+), 51 deletions(-)
>
>diff --git a/doc/guides/prog_guide/hash_lib.rst b/doc/guides/prog_guide/hash_lib.rst
>index 85a6edfa8b16..b00446e949ba 100644
>--- a/doc/guides/prog_guide/hash_lib.rst
>+++ b/doc/guides/prog_guide/hash_lib.rst
>@@ -108,8 +108,7 @@ Extendable Bucket Functionality support
> An extra flag is used to enable this functionality (flag is not set by default). When the (RTE_HASH_EXTRA_FLAGS_EXT_TABLE) is set
>and
> in the very unlikely case due to excessive hash collisions that a key has failed to be inserted, the hash table bucket is extended with a
>linked
> list to insert these failed keys. This feature is important for the workloads (e.g. telco workloads) that need to insert up to 100% of the
>-hash table size and can't tolerate any key insertion failure (even if very few). Currently the extendable bucket is not supported
>-with the lock-free concurrency implementation (RTE_HASH_EXTRA_FLAGS_RW_CONCURRENCY_LF).
>+hash table size and can't tolerate any key insertion failure (even if very few).
>
>
> Implementation Details (non Extendable Bucket Case)
>diff --git a/lib/librte_hash/rte_cuckoo_hash.c b/lib/librte_hash/rte_cuckoo_hash.c
>index c01489ba5193..54762533ce36 100644
>--- a/lib/librte_hash/rte_cuckoo_hash.c
>+++ b/lib/librte_hash/rte_cuckoo_hash.c
>@@ -170,15 +170,6 @@ rte_hash_create(const struct rte_hash_parameters *params)
> return NULL;
> }
>
>- if ((params->extra_flag & RTE_HASH_EXTRA_FLAGS_RW_CONCURRENCY_LF) &&
>- (params->extra_flag & RTE_HASH_EXTRA_FLAGS_EXT_TABLE)) {
>- rte_errno = EINVAL;
>- RTE_LOG(ERR, HASH, "rte_hash_create: extendable bucket "
>- "feature not supported with rw concurrency "
>- "lock free\n");
>- return NULL;
>- }
>-
> /* Check extra flags field to check extra options. */
> if (params->extra_flag & RTE_HASH_EXTRA_FLAGS_TRANS_MEM_SUPPORT)
> hw_trans_mem_support = 1;
>@@ -1054,7 +1045,15 @@ __rte_hash_add_key_with_hash(const struct rte_hash *h, const void *key,
> /* Check if slot is available */
> if (likely(cur_bkt->key_idx[i] == EMPTY_SLOT)) {
> cur_bkt->sig_current[i] = short_sig;
>- cur_bkt->key_idx[i] = new_idx;
>+ /* Key can be of arbitrary length, so it is
>+ * not possible to store it atomically.
>+ * Hence the new key element's memory stores
>+ * (key as well as data) should be complete
>+ * before it is referenced.
>+ */
>+ __atomic_store_n(&cur_bkt->key_idx[i],
>+ new_idx,
>+ __ATOMIC_RELEASE);
> __hash_rw_writer_unlock(h);
> return new_idx - 1;
> }
>@@ -1072,10 +1071,23 @@ __rte_hash_add_key_with_hash(const struct rte_hash *h, const void *key,
> bkt_id = (uint32_t)((uintptr_t)ext_bkt_id) - 1;
> /* Use the first location of the new bucket */
> (h->buckets_ext[bkt_id]).sig_current[0] = short_sig;
>- (h->buckets_ext[bkt_id]).key_idx[0] = new_idx;
>+ /* Key can be of arbitrary length, so it is
>+ * not possible to store it atomically.
>+ * Hence the new key element's memory stores
>+ * (key as well as data) should be complete
>+ * before it is referenced.
>+ */
>+ __atomic_store_n(&(h->buckets_ext[bkt_id]).key_idx[0],
>+ new_idx,
>+ __ATOMIC_RELEASE);
[Wang, Yipeng] Since it has not been linked and later on the linking is protected by
release, do we really need atomic store here?
> /* Link the new bucket to sec bucket linked list */
> last = rte_hash_get_last_bkt(sec_bkt);
>- last->next = &h->buckets_ext[bkt_id];
>+ /* New bucket's memory stores (key as well as data)
>+ * should be complete before it is referenced
>+ */
>+ __atomic_store_n(&last->next,
>+ &h->buckets_ext[bkt_id],
>+ __ATOMIC_RELEASE);
> __hash_rw_writer_unlock(h);
> return new_idx - 1;
>
>@@ -1366,7 +1378,8 @@ remove_entry(const struct rte_hash *h, struct rte_hash_bucket *bkt, unsigned i)
> * empty slot.
> */
> static inline void
>-__rte_hash_compact_ll(struct rte_hash_bucket *cur_bkt, int pos) {
>+__rte_hash_compact_ll(const struct rte_hash *h,
>+ struct rte_hash_bucket *cur_bkt, int pos) {
> int i;
> struct rte_hash_bucket *last_bkt;
>
>@@ -1377,10 +1390,27 @@ __rte_hash_compact_ll(struct rte_hash_bucket *cur_bkt, int pos) {
>
> for (i = RTE_HASH_BUCKET_ENTRIES - 1; i >= 0; i--) {
> if (last_bkt->key_idx[i] != EMPTY_SLOT) {
>- cur_bkt->key_idx[pos] = last_bkt->key_idx[i];
> cur_bkt->sig_current[pos] = last_bkt->sig_current[i];
>+ __atomic_store_n(&cur_bkt->key_idx[pos],
>+ last_bkt->key_idx[i],
>+ __ATOMIC_RELEASE);
>+ if (h->readwrite_concur_lf_support) {
>+ /* Inform the readers that the table has changed
>+ * Since there is one writer, load acquires on
>+ * tbl_chng_cnt are not required.
>+ */
>+ __atomic_store_n(h->tbl_chng_cnt,
>+ *h->tbl_chng_cnt + 1,
>+ __ATOMIC_RELEASE);
>+ /* The stores to sig_alt and sig_current should
>+ * not move above the store to tbl_chng_cnt.
>+ */
>+ __atomic_thread_fence(__ATOMIC_RELEASE);
>+ }
> last_bkt->sig_current[i] = NULL_SIGNATURE;
>- last_bkt->key_idx[i] = EMPTY_SLOT;
>+ __atomic_store_n(&last_bkt->key_idx[i],
>+ EMPTY_SLOT,
>+ __ATOMIC_RELEASE);
> return;
> }
> }
>@@ -1449,7 +1479,7 @@ __rte_hash_del_key_with_hash(const struct rte_hash *h, const void *key,
> /* look for key in primary bucket */
> ret = search_and_remove(h, key, prim_bkt, short_sig, &pos);
> if (ret != -1) {
>- __rte_hash_compact_ll(prim_bkt, pos);
>+ __rte_hash_compact_ll(h, prim_bkt, pos);
> last_bkt = prim_bkt->next;
> prev_bkt = prim_bkt;
> goto return_bkt;
>@@ -1461,7 +1491,7 @@ __rte_hash_del_key_with_hash(const struct rte_hash *h, const void *key,
> FOR_EACH_BUCKET(cur_bkt, sec_bkt) {
> ret = search_and_remove(h, key, cur_bkt, short_sig, &pos);
> if (ret != -1) {
>- __rte_hash_compact_ll(cur_bkt, pos);
>+ __rte_hash_compact_ll(h, cur_bkt, pos);
> last_bkt = sec_bkt->next;
> prev_bkt = sec_bkt;
> goto return_bkt;
>@@ -1488,11 +1518,21 @@ __rte_hash_del_key_with_hash(const struct rte_hash *h, const void *key,
> }
> /* found empty bucket and recycle */
> if (i == RTE_HASH_BUCKET_ENTRIES) {
>- prev_bkt->next = last_bkt->next = NULL;
>+ __atomic_store_n(&prev_bkt->next,
>+ NULL,
>+ __ATOMIC_RELEASE);
> uint32_t index = last_bkt - h->buckets_ext + 1;
>- rte_ring_sp_enqueue(h->free_ext_bkts, (void *)(uintptr_t)index);
>- }
>+ if (!h->no_free_on_del)
>+ rte_ring_sp_enqueue(h->free_ext_bkts, (void *)(uintptr_t)index);
>+ else {
>+ struct rte_hash_key *key_slot =
>+ (struct rte_hash_key *)(
>+ (char *)h->key_store +
>+ ret * h->key_entry_size);
>+ key_slot->ext_bkt_to_free = index;
[Wang, Yipeng] Is there chance that a key_slot may already have one previous ext_bkt
and now got overwritten, so that the previous one gone forever?
>
>+ }
>+ }
> __hash_rw_writer_unlock(h);
> return ret;
> }
>@@ -1567,6 +1607,14 @@ rte_hash_free_key_with_position(const struct rte_hash *h,
> (void *)((uintptr_t)position));
> }
>
>+ const struct rte_hash_key *key_slot = (const struct rte_hash_key *)(
>+ (const char *)h->key_store +
>+ position * h->key_entry_size);
>+ uint32_t index = key_slot->ext_bkt_to_free;
>+ if (!index)
>+ /* Recycle empty ext bkt to free list. */
>+ rte_ring_sp_enqueue(h->free_ext_bkts, (void *)(uintptr_t)index);
>+
> return 0;
> }
>
>@@ -1855,6 +1903,9 @@ __rte_hash_lookup_bulk_lf(const struct rte_hash *h, const void **keys,
> rte_prefetch0(secondary_bkt[i]);
> }
>
>+ for (i = 0; i < num_keys; i++)
>+ positions[i] = -ENOENT;
[Wang, Yipeng] So is this for performance reason?
>+
> do {
> /* Load the table change counter before the lookup
> * starts. Acquire semantics will make sure that
>@@ -1899,7 +1950,6 @@ __rte_hash_lookup_bulk_lf(const struct rte_hash *h, const void **keys,
>
> /* Compare keys, first hits in primary first */
> for (i = 0; i < num_keys; i++) {
>- positions[i] = -ENOENT;
> while (prim_hitmask[i]) {
> uint32_t hit_index =
> __builtin_ctzl(prim_hitmask[i])
>@@ -1972,6 +2022,36 @@ __rte_hash_lookup_bulk_lf(const struct rte_hash *h, const void **keys,
> continue;
> }
>
>+
>+ /* all found, do not need to go through ext bkt */
>+ if (hits == ((1ULL << num_keys) - 1)) {
>+ if (hit_mask != NULL)
>+ *hit_mask = hits;
[Wang, Yipeng] I think you need to check the version counter before return,
and how about the fence?
>+ return;
>+ }
>+ /* need to check ext buckets for match */
>+ if (h->ext_table_support) {
>+ for (i = 0; i < num_keys; i++) {
>+ if ((hits & (1ULL << i)) != 0)
>+ continue;
>+ next_bkt = secondary_bkt[i]->next;
>+ FOR_EACH_BUCKET(cur_bkt, next_bkt) {
>+ if (data != NULL)
>+ ret = search_one_bucket_lf(h,
>+ keys[i], sig[i],
>+ &data[i], cur_bkt);
>+ else
>+ ret = search_one_bucket_lf(h,
>+ keys[i], sig[i],
>+ NULL, cur_bkt);
>+ if (ret != -1) {
>+ positions[i] = ret;
>+ hits |= 1ULL << i;
>+ break;
>+ }
>+ }
>+ }
>+ }
> /* The loads of sig_current in compare_signatures
> * should not move below the load from tbl_chng_cnt.
> */
>@@ -1988,34 +2068,6 @@ __rte_hash_lookup_bulk_lf(const struct rte_hash *h, const void **keys,
> __ATOMIC_ACQUIRE);
> } while (cnt_b != cnt_a);
>
>- /* all found, do not need to go through ext bkt */
>- if ((hits == ((1ULL << num_keys) - 1)) || !h->ext_table_support) {
>- if (hit_mask != NULL)
>- *hit_mask = hits;
>- __hash_rw_reader_unlock(h);
>- return;
>- }
>-
>- /* need to check ext buckets for match */
>- for (i = 0; i < num_keys; i++) {
>- if ((hits & (1ULL << i)) != 0)
>- continue;
>- next_bkt = secondary_bkt[i]->next;
>- FOR_EACH_BUCKET(cur_bkt, next_bkt) {
>- if (data != NULL)
>- ret = search_one_bucket_lf(h, keys[i],
>- sig[i], &data[i], cur_bkt);
>- else
>- ret = search_one_bucket_lf(h, keys[i],
>- sig[i], NULL, cur_bkt);
>- if (ret != -1) {
>- positions[i] = ret;
>- hits |= 1ULL << i;
>- break;
>- }
>- }
>- }
>-
> if (hit_mask != NULL)
> *hit_mask = hits;
> }
>diff --git a/lib/librte_hash/rte_cuckoo_hash.h b/lib/librte_hash/rte_cuckoo_hash.h
>index eacdaa8d4684..062cc2dd0296 100644
>--- a/lib/librte_hash/rte_cuckoo_hash.h
>+++ b/lib/librte_hash/rte_cuckoo_hash.h
>@@ -129,6 +129,14 @@ struct lcore_cache {
>
> /* Structure that stores key-value pair */
> struct rte_hash_key {
>+ /* Stores index of an empty ext bkt to be recycled on calling
>+ * rte_hash_del_xxx APIs. When lock free read-wrie concurrency is
[Wang, Yipeng] typo
>+ * enabled, an empty ext bkt cannot be put into free list immediately
>+ * (as readers might be using it still). Hence freeing of the ext bkt
>+ * is piggy-backed to freeing of the key index.
>+ */
[Wang, Yipeng] I am thinking if this breaks the "guarantee" provided by ext table, Since
a whole bucket could not be reused if one key not freed. Is there any fundamental issue with
a new API to recycle ext bucket or you just do not want to add a new API?
>+ uint32_t ext_bkt_to_free;
>+
> union {
> uintptr_t idata;
> void *pdata;
>--
>2.17.1
More information about the dev
mailing list