Add WIP bulk functions for DICT/OA

This commit is contained in:
Patrick Pelissier committed 2024-05-09 14:10:22 +02:00
1 parent 8fd79f07d5
commit 22d62f94bb
2 files changed
+115 -2

No files matched your search

+12 -1
View File
@@ -143,6 +143,13 @@
#endif
/* Prefetch . */
#if defined(__GNUC__)
# define M_PREFETCH(p) __builtin_prefetch((p), 0, 0)
#else
# define M_PREFETCH(p) (void) 0
#endif
/* Ignore some warnings detected by some compilers in the library.
* Whatever we do, there is some warnings that cannot be fixed.
* So they are ignored in order to avoid polluting the user with
@@ -2895,11 +2902,15 @@ M_PARSE_DEFAULT_TYPE_DEF(m_core_parse_ldouble, long double, strtold, )
/* C++ doesn't support flexible array within a structure.
Let's define at least one element for an array. */
Let's define at least one element for an array.
It doesn't also support VLA usage in function prototype.
*/
#ifdef __cplusplus
# define M_MIN_FLEX_ARRAY_SIZE 1
# define M_VLA(n)
#else
# define M_MIN_FLEX_ARRAY_SIZE
# define M_VLA(n) n
#endif
#if M_USE_STDARG && M_USE_STDIO
+103 -1
View File
@@ -2021,9 +2021,111 @@ enum m_d1ct_oa_element_e {
return M_CONST_CAST(it_deref_t, M_F(name, _ref)(it)); \
} \
\
M_D1CT_FUNC_ADDITIONAL_DEF2(name, key_type, key_oplist, value_type, value_oplist, isSet, dict_t, dict_it_t, it_deref_t)
M_D1CT_FUNC_ADDITIONAL_DEF2(name, key_type, key_oplist, value_type, value_oplist, isSet, dict_t, dict_it_t, it_deref_t) \
M_IF(isSet)(M_EAT, M_D1CT_OA_DEF_BULK)(name, key_type, key_oplist, value_type, value_oplist, isSet, coeff_down, coeff_up, dict_t, dict_it_t, it_deref_t)
// Number of loads to perform in advance
#ifndef M_USE_MAX_PREFETCH
#define M_USE_MAX_PREFETCH 16
#endif
// WIP. Only if isSet is true for the time being
#define M_D1CT_OA_DEF_BULK(name, key_type, key_oplist, value_type, value_oplist, isSet, coeff_down, coeff_up, dict_t, dict_it_t, it_deref_t) \
\
M_INLINE void \
M_F(name, _bulk_get)(unsigned n, value_type val[M_VLA(n)], dict_t dict, key_type const key[M_VLA(n)], value_type def) \
{ \
M_D1CT_OA_CONTRACT (dict); \
size_t h[M_USE_MAX_PREFETCH]; \
M_F(name, _pair_ct) *data = dict->data; \
const size_t mask = dict->mask; \
const unsigned num = M_MIN(M_USE_MAX_PREFETCH, n); \
for(unsigned i = 0 ; i < num; i++) { \
h[i] = M_CALL_HASH(key_oplist, key[i]) & mask; \
M_PREFETCH(&data[h[i]]); \
} \
for (unsigned i = 0; i < n; i++) { \
size_t s = 1, p = h[i % M_USE_MAX_PREFETCH]; \
if (i < n-M_USE_MAX_PREFETCH) { \
h[i % M_USE_MAX_PREFETCH] = M_CALL_HASH(key_oplist, key[i+M_USE_MAX_PREFETCH]) & mask; \
M_PREFETCH(&data[h[i % M_USE_MAX_PREFETCH]]); \
} \
while (true) { \
if (M_LIKELY (M_CALL_EQUAL(key_oplist, data[p].key, key[i]))) { \
M_CALL_SET(value_oplist, val[i], data[p].value); \
break; \
} \
if (M_LIKELY (M_CALL_OOR_EQUAL (key_oplist, data[p].key, M_D1CT_OA_EMPTY)) ) { \
M_CALL_SET(value_oplist, val[i], def); \
break; \
} \
p = (p + M_D1CT_OA_PROBING(s)) & mask; \
M_ASSERT(p <= mask); \
} \
} \
} \
\
M_INLINE void \
M_F(name, _bulk_set)(dict_t dict, unsigned n, value_type const val[M_VLA(n)], key_type const key[M_VLA(n)]) \
{ \
M_D1CT_OA_CONTRACT (dict); \
size_t h[M_USE_MAX_PREFETCH]; \
const unsigned num = M_MIN(M_USE_MAX_PREFETCH, n); \
for(unsigned i = 0 ; i < num; i++) { \
h[i] = M_CALL_HASH(key_oplist, key[i]); \
M_PREFETCH(&dict->data[h[i] & dict->mask]); \
} \
\
for (unsigned i = 0; i < n; i++) { \
size_t p = h[i % M_USE_MAX_PREFETCH] & dict->mask; \
if (i < n-M_USE_MAX_PREFETCH) { \
h[i % M_USE_MAX_PREFETCH] = M_CALL_HASH(key_oplist, key[i+M_USE_MAX_PREFETCH]); \
M_PREFETCH(&dict->data[h[i % M_USE_MAX_PREFETCH] & dict->mask]); \
} \
if (M_UNLIKELY (M_CALL_EQUAL(key_oplist, dict->data[p].key, key[i]))) { \
M_CALL_SET(value_oplist, dict->data[p].value, val[i]); \
continue; \
} \
/* Search loop for insertion */ \
size_t s = 1; \
size_t delPos = SIZE_MAX; \
while (M_UNLIKELY (!M_CALL_OOR_EQUAL (key_oplist, dict->data[p].key, M_D1CT_OA_EMPTY)) ) { \
if (M_CALL_EQUAL(key_oplist, dict->data[p].key, key[i])) { \
M_CALL_SET(value_oplist, dict->data[p].value, val[i]); \
goto next; \
} \
if (M_CALL_OOR_EQUAL (key_oplist, dict->data[p].key, M_D1CT_OA_DELETED) \
&& (delPos == SIZE_MAX)) \
delPos = p; \
p = (p + M_D1CT_OA_PROBING(s)) & dict->mask; \
} \
/* Add new item */ \
if (delPos != SIZE_MAX) { \
p = delPos; \
dict->count_delete --; \
} \
M_CALL_INIT_SET(key_oplist, dict->data[p].key, key[i]); \
M_CALL_INIT_SET(value_oplist, dict->data[p].value, val[i]); \
dict->count++; \
dict->count_delete ++; \
\
if (M_UNLIKELY (dict->count_delete >= dict->upper_limit)) { \
size_t newSize = dict->mask+1; \
if (dict->count > (dict->mask / 2)) { \
newSize += newSize; \
if (M_UNLIKELY_NOMEM (newSize <= dict->mask+1)) { \
M_MEMORY_FULL((size_t)-1); \
} \
} \
M_C3(m_d1ct_,name,_resize_up)(dict, newSize, true); \
} \
next: \
(void)0; \
} \
M_D1CT_OA_CONTRACT (dict); \
} \
/******************************** INTERNAL ***********************************/
#if M_USE_SMALL_NAME