|
|
@@ -0,0 +1,275 @@
|
|
|
+#ifndef __STARPU_STATIC_INLINE_BITMAP_H__
|
|
|
+#define __STARPU_STATIC_INLINE_BITMAP_H__
|
|
|
+
|
|
|
+#include <common/utils.h>
|
|
|
+
|
|
|
+#ifdef __cplusplus
|
|
|
+extern "C"
|
|
|
+{
|
|
|
+#endif
|
|
|
+
|
|
|
+#ifndef LONG_BIT
|
|
|
+#define LONG_BIT (sizeof(unsigned long) * 8)
|
|
|
+#endif
|
|
|
+
|
|
|
+#define BITMAP_SIZE ((STARPU_NMAXWORKERS - 1)/LONG_BIT) + 1
|
|
|
+
|
|
|
+/** */
|
|
|
+static inline struct starpu_bitmap *starpu_bitmap_create(void) STARPU_ATTRIBUTE_MALLOC;
|
|
|
+/** free \p b */
|
|
|
+static inline void starpu_bitmap_destroy(struct starpu_bitmap *b);
|
|
|
+
|
|
|
+/** set bit \p e in \p b */
|
|
|
+static inline void starpu_bitmap_set(struct starpu_bitmap *b, int e);
|
|
|
+/** unset bit \p e in \p b */
|
|
|
+static inline void starpu_bitmap_unset(struct starpu_bitmap *b, int e);
|
|
|
+/** unset all bits in \p b */
|
|
|
+static inline void starpu_bitmap_unset_all(struct starpu_bitmap *b);
|
|
|
+
|
|
|
+/** return true iff bit \p e is set in \p b */
|
|
|
+static inline int starpu_bitmap_get(struct starpu_bitmap *b, int e);
|
|
|
+/** Basically compute \c starpu_bitmap_unset_all(\p a) ; \p a = \p b & \p c; */
|
|
|
+static inline void starpu_bitmap_unset_and(struct starpu_bitmap *a, struct starpu_bitmap *b, struct starpu_bitmap *c);
|
|
|
+/** Basically compute \p a |= \p b */
|
|
|
+static inline void starpu_bitmap_or(struct starpu_bitmap *a, struct starpu_bitmap *b);
|
|
|
+/** return 1 iff \p e is set in \p b1 AND \p e is set in \p b2 */
|
|
|
+static inline int starpu_bitmap_and_get(struct starpu_bitmap *b1, struct starpu_bitmap *b2, int e);
|
|
|
+/** return the number of set bits in \p b */
|
|
|
+static inline int starpu_bitmap_cardinal(struct starpu_bitmap *b);
|
|
|
+
|
|
|
+/** return the index of the first set bit of \p b, -1 if none */
|
|
|
+static inline int starpu_bitmap_first(struct starpu_bitmap *b);
|
|
|
+/** return the position of the last set bit of \p b, -1 if none */
|
|
|
+static inline int starpu_bitmap_last(struct starpu_bitmap *b);
|
|
|
+/** return the position of set bit right after \p e in \p b, -1 if none */
|
|
|
+static inline int starpu_bitmap_next(struct starpu_bitmap *b, int e);
|
|
|
+/** todo */
|
|
|
+static inline int starpu_bitmap_has_next(struct starpu_bitmap *b, int e);
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+struct starpu_bitmap
|
|
|
+{
|
|
|
+ unsigned long bits[BITMAP_SIZE];
|
|
|
+ int cardinal;
|
|
|
+};
|
|
|
+
|
|
|
+#ifdef DEBUG_BITMAP
|
|
|
+static int check_bitmap(struct starpu_bitmap *b)
|
|
|
+{
|
|
|
+ int card = b->cardinal;
|
|
|
+ int i = starpu_bitmap_first(b);
|
|
|
+ int j;
|
|
|
+ for(j = 0; j < card; j++)
|
|
|
+ {
|
|
|
+ if(i == -1)
|
|
|
+ return 0;
|
|
|
+ int tmp = starpu_bitmap_next(b,i);
|
|
|
+ if(tmp == i)
|
|
|
+ return 0;
|
|
|
+ i = tmp;
|
|
|
+ }
|
|
|
+ if(i != -1)
|
|
|
+ return 0;
|
|
|
+ return 1;
|
|
|
+}
|
|
|
+#else
|
|
|
+#define check_bitmap(b) 1
|
|
|
+#endif
|
|
|
+
|
|
|
+static int _count_bit_static(unsigned long e)
|
|
|
+{
|
|
|
+#if (__GNUC__ >= 4) || ((__GNUC__ == 3) && (__GNUC_MINOR__) >= 4)
|
|
|
+ return __builtin_popcountl(e);
|
|
|
+#else
|
|
|
+ int c = 0;
|
|
|
+ while(e)
|
|
|
+ {
|
|
|
+ c += e&1;
|
|
|
+ e >>= 1;
|
|
|
+ }
|
|
|
+ return c;
|
|
|
+#endif
|
|
|
+}
|
|
|
+
|
|
|
+static inline struct starpu_bitmap *starpu_bitmap_create()
|
|
|
+{
|
|
|
+ struct starpu_bitmap *b;
|
|
|
+ _STARPU_CALLOC(b, 1, sizeof(*b));
|
|
|
+ for(int i=0 ; i < BITMAP_SIZE; i++)
|
|
|
+ b->bits[i] = 0;
|
|
|
+ return b;
|
|
|
+}
|
|
|
+
|
|
|
+static inline void starpu_bitmap_destroy(struct starpu_bitmap * b)
|
|
|
+{
|
|
|
+ if(b)
|
|
|
+ {
|
|
|
+ free(b);
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+static inline void starpu_bitmap_set(struct starpu_bitmap * b, int e)
|
|
|
+{
|
|
|
+ if(!starpu_bitmap_get(b, e))
|
|
|
+ b->cardinal++;
|
|
|
+ else
|
|
|
+ return;
|
|
|
+ STARPU_ASSERT(e/LONG_BIT < BITMAP_SIZE);
|
|
|
+ b->bits[e/LONG_BIT] |= (1ul << (e%LONG_BIT));
|
|
|
+ STARPU_ASSERT(check_bitmap(b));
|
|
|
+}
|
|
|
+static inline void starpu_bitmap_unset(struct starpu_bitmap *b, int e)
|
|
|
+{
|
|
|
+ if(starpu_bitmap_get(b, e))
|
|
|
+ b->cardinal--;
|
|
|
+ else
|
|
|
+ return;
|
|
|
+ STARPU_ASSERT(e/LONG_BIT < BITMAP_SIZE);
|
|
|
+ if(e / LONG_BIT > BITMAP_SIZE)
|
|
|
+ return;
|
|
|
+ b->bits[e/LONG_BIT] &= ~(1ul << (e%LONG_BIT));
|
|
|
+ STARPU_ASSERT(check_bitmap(b));
|
|
|
+}
|
|
|
+
|
|
|
+static inline void starpu_bitmap_unset_all(struct starpu_bitmap * b)
|
|
|
+{
|
|
|
+ memset(b->bits, 0, BITMAP_SIZE * sizeof(unsigned long));
|
|
|
+}
|
|
|
+
|
|
|
+static inline void starpu_bitmap_unset_and(struct starpu_bitmap * a, struct starpu_bitmap * b, struct starpu_bitmap * c)
|
|
|
+{
|
|
|
+ a->cardinal = 0;
|
|
|
+ int i;
|
|
|
+ for(i = 0; i < BITMAP_SIZE; i++)
|
|
|
+ {
|
|
|
+ a->bits[i] = b->bits[i] & c->bits[i];
|
|
|
+ a->cardinal += _count_bit_static(a->bits[i]);
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+static inline int starpu_bitmap_get(struct starpu_bitmap * b, int e)
|
|
|
+{
|
|
|
+ STARPU_ASSERT(e / LONG_BIT < BITMAP_SIZE);
|
|
|
+ if(e / LONG_BIT >= BITMAP_SIZE)
|
|
|
+ return 0;
|
|
|
+ return (b->bits[e/LONG_BIT] & (1ul << (e%LONG_BIT))) ?
|
|
|
+ 1:
|
|
|
+ 0;
|
|
|
+}
|
|
|
+
|
|
|
+static inline void starpu_bitmap_or(struct starpu_bitmap * a, struct starpu_bitmap * b)
|
|
|
+{
|
|
|
+ int i;
|
|
|
+ a->cardinal = 0;
|
|
|
+ for(i = 0; i < BITMAP_SIZE; i++)
|
|
|
+ {
|
|
|
+ a->bits[i] |= b->bits[i];
|
|
|
+ a->cardinal += _count_bit_static(a->bits[i]);
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+int starpu_bitmap_and_get(struct starpu_bitmap * b1, struct starpu_bitmap * b2, int e)
|
|
|
+{
|
|
|
+ return starpu_bitmap_get(b1,e) && starpu_bitmap_get(b2,e);
|
|
|
+}
|
|
|
+
|
|
|
+static inline int starpu_bitmap_cardinal(struct starpu_bitmap * b)
|
|
|
+{
|
|
|
+ return b->cardinal;
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+static inline int get_first_bit_rank(unsigned long ms)
|
|
|
+{
|
|
|
+ STARPU_ASSERT(ms != 0);
|
|
|
+#if (__GNUC__ >= 4) || ((__GNUC__ == 3) && (__GNUC_MINOR__ >= 4))
|
|
|
+ return __builtin_ffsl(ms) - 1;
|
|
|
+#else
|
|
|
+ unsigned long m = 1ul;
|
|
|
+ int i = 0;
|
|
|
+ while(!(m&ms))
|
|
|
+ i++,m<<=1;
|
|
|
+ return i;
|
|
|
+#endif
|
|
|
+}
|
|
|
+
|
|
|
+static inline int get_last_bit_rank(unsigned long l)
|
|
|
+{
|
|
|
+ STARPU_ASSERT(l != 0);
|
|
|
+#if (__GNUC__ >= 4) || ((__GNUC__ == 3) && (__GNUC_MINOR__ >= 4))
|
|
|
+ return 8*sizeof(l) - __builtin_clzl(l);
|
|
|
+#else
|
|
|
+ int ibit = LONG_BIT - 1;
|
|
|
+ while((!(1ul << ibit)) & l)
|
|
|
+ ibit--;
|
|
|
+ STARPU_ASSERT(ibit >= 0);
|
|
|
+ return ibit;
|
|
|
+#endif
|
|
|
+}
|
|
|
+
|
|
|
+static inline int starpu_bitmap_first(struct starpu_bitmap * b)
|
|
|
+{
|
|
|
+ int i = 0;
|
|
|
+ while(i < BITMAP_SIZE && !b->bits[i])
|
|
|
+ i++;
|
|
|
+ if( i == BITMAP_SIZE)
|
|
|
+ return -1;
|
|
|
+ int nb_long = i;
|
|
|
+ unsigned long ms = b->bits[i];
|
|
|
+
|
|
|
+ return (nb_long * LONG_BIT) + get_first_bit_rank(ms);
|
|
|
+}
|
|
|
+
|
|
|
+static inline int starpu_bitmap_has_next(struct starpu_bitmap * b, int e)
|
|
|
+{
|
|
|
+ int nb_long = (e+1) / LONG_BIT;
|
|
|
+ int nb_bit = (e+1) % LONG_BIT;
|
|
|
+ unsigned long mask = (~0ul) << nb_bit;
|
|
|
+ if(b->bits[nb_long] & mask)
|
|
|
+ return 1;
|
|
|
+ for(nb_long++; nb_long < BITMAP_SIZE; nb_long++)
|
|
|
+ if(b->bits[nb_long])
|
|
|
+ return 1;
|
|
|
+ return 0;
|
|
|
+}
|
|
|
+
|
|
|
+static inline int starpu_bitmap_last(struct starpu_bitmap * b)
|
|
|
+{
|
|
|
+ if(b->cardinal == 0)
|
|
|
+ return -1;
|
|
|
+ int ilong;
|
|
|
+ for(ilong = BITMAP_SIZE - 1; ilong >= 0; ilong--)
|
|
|
+ {
|
|
|
+ if(b->bits[ilong])
|
|
|
+ break;
|
|
|
+ }
|
|
|
+ STARPU_ASSERT(ilong >= 0);
|
|
|
+ unsigned long l = b->bits[ilong];
|
|
|
+ return ilong * LONG_BIT + get_last_bit_rank(l);
|
|
|
+}
|
|
|
+
|
|
|
+static inline int starpu_bitmap_next(struct starpu_bitmap *b, int e)
|
|
|
+{
|
|
|
+ int nb_long = e / LONG_BIT;
|
|
|
+ int nb_bit = e % LONG_BIT;
|
|
|
+ unsigned long rest = nb_bit == LONG_BIT - 1 ? 0 : (~0ul << (nb_bit + 1)) & b->bits[nb_long];
|
|
|
+ if(nb_bit != (LONG_BIT - 1) && rest)
|
|
|
+ {
|
|
|
+ int i = get_first_bit_rank(rest);
|
|
|
+ STARPU_ASSERT(i >= 0 && i < LONG_BIT);
|
|
|
+ return (nb_long * LONG_BIT) + i;
|
|
|
+ }
|
|
|
+
|
|
|
+ for(nb_long++;nb_long < BITMAP_SIZE; nb_long++)
|
|
|
+ if(b->bits[nb_long])
|
|
|
+ return nb_long * LONG_BIT + get_first_bit_rank(b->bits[nb_long]);
|
|
|
+ return -1;
|
|
|
+}
|
|
|
+
|
|
|
+#ifdef __cplusplus
|
|
|
+}
|
|
|
+#endif
|
|
|
+
|
|
|
+#endif /* __STARPU_STATIC_INLINE_BITMAP_H__ */
|
