benchmark/ms-queue/my_queue.h

   1 #include <stdatomic.h>
   2
   3 #define MAX_NODES                       0xf
   4
   5 typedef unsigned long long pointer;
   6 typedef atomic_ullong pointer_t;
   7
   8 #define MAKE_POINTER(ptr, count)        ((((pointer)count) << 32) | ptr)
   9 #define PTR_MASK 0xffffffffLL
  10 #define COUNT_MASK (0xffffffffLL << 32)
  11
  12 static inline void set_count(pointer *p, unsigned int val) { *p = (*p & ~COUNT_MASK) | ((pointer)val << 32); }
  13 static inline void set_ptr(pointer *p, unsigned int val) { *p = (*p & ~PTR_MASK) | val; }
  14 static inline unsigned int get_count(pointer p) { return (p & COUNT_MASK) >> 32; }
  15 static inline unsigned int get_ptr(pointer p) { return p & PTR_MASK; }
  16
  17 typedef struct node {
  18         unsigned int value;
  19         pointer_t next;
  20 } node_t;
  21
  22 typedef struct {
  23         pointer_t head;
  24         pointer_t tail;
  25         node_t nodes[MAX_NODES + 1];
  26 } queue_t;
  27
  28 void init_queue(queue_t *q, int num_threads);
  29 void enqueue(queue_t *q, unsigned int val);
  30 unsigned int dequeue(queue_t *q);
  31 int get_thread_num();