| [f7f07f6] | 1 | #include <stdio.h>
 | 
|---|
 | 2 | #include "locks.hfa"
 | 
|---|
 | 3 | #include <stdlib.hfa>
 | 
|---|
 | 4 | #include <thread.hfa>
 | 
|---|
 | 5 | #include <containers/array.hfa>
 | 
|---|
 | 6 | 
 | 
|---|
 | 7 | static unsigned int taskCount = 4;
 | 
|---|
 | 8 | static unsigned int threadCount = 2;
 | 
|---|
 | 9 | static unsigned int lockCount = 1;
 | 
|---|
| [cf444b6] | 10 | static unsigned int total_times = 320000;
 | 
|---|
 | 11 | static unsigned int num_times;
 | 
|---|
| [f7f07f6] | 12 | static const int workBufferSize = 16;
 | 
|---|
 | 13 | static unsigned int work_unlocked = 10000;
 | 
|---|
 | 14 | static unsigned int work_locked = 10000;
 | 
|---|
 | 15 | 
 | 
|---|
 | 16 | // taken from martin's thread_test
 | 
|---|
 | 17 | static inline void dowork(volatile int* buffer, unsigned int steps) {
 | 
|---|
 | 18 |   int value = 0;
 | 
|---|
 | 19 |   for (unsigned int i = 0; i < steps; i += 1) {
 | 
|---|
 | 20 |     // a little more work than just a single memory access helps with stability
 | 
|---|
 | 21 |     value += (buffer[i % workBufferSize] * 17) / 23 + 55;
 | 
|---|
 | 22 |   }
 | 
|---|
 | 23 |   buffer[0] += value;
 | 
|---|
 | 24 | }
 | 
|---|
 | 25 | 
 | 
|---|
 | 26 | thread worker {
 | 
|---|
 | 27 |     linear_backoff_then_block_lock * locks;
 | 
|---|
| [cf444b6] | 28 |     bool improved;
 | 
|---|
| [f7f07f6] | 29 | };
 | 
|---|
 | 30 | 
 | 
|---|
| [cf444b6] | 31 | void ?{}( worker & w, linear_backoff_then_block_lock * locks, bool improved ) {
 | 
|---|
| [f7f07f6] | 32 |         w.locks = locks;
 | 
|---|
| [cf444b6] | 33 |     w.improved = improved;
 | 
|---|
| [f7f07f6] | 34 | }
 | 
|---|
 | 35 | 
 | 
|---|
 | 36 | 
 | 
|---|
 | 37 | void main( worker & this ) with(this) {
 | 
|---|
 | 38 |         int buffer[workBufferSize];
 | 
|---|
 | 39 |     for (int i = 0; i < workBufferSize; i += 1) buffer[i] = rand() % 1024;
 | 
|---|
 | 40 |     unsigned int lck = rand() % lockCount;
 | 
|---|
| [cf444b6] | 41 |     linear_backoff_then_block_lock * curr_lock = &locks[lck];
 | 
|---|
| [f7f07f6] | 42 |     for (unsigned int i = 0; i < num_times; i++) {
 | 
|---|
 | 43 |         dowork(buffer, work_unlocked);
 | 
|---|
| [cf444b6] | 44 |         if (improved) lock_improved(*curr_lock);
 | 
|---|
 | 45 |         else lock(*curr_lock);
 | 
|---|
| [f7f07f6] | 46 |         dowork(buffer, work_locked);
 | 
|---|
| [cf444b6] | 47 |         unlock(*curr_lock);
 | 
|---|
| [f7f07f6] | 48 |         lck = rand() % lockCount;
 | 
|---|
| [cf444b6] | 49 |         curr_lock = &locks[lck];
 | 
|---|
| [f7f07f6] | 50 |     }
 | 
|---|
 | 51 | }
 | 
|---|
 | 52 | 
 | 
|---|
| [cf444b6] | 53 | 
 | 
|---|
| [f7f07f6] | 54 | int main(int argc, char* argv[]) {
 | 
|---|
 | 55 |     switch (argc) {
 | 
|---|
| [cf444b6] | 56 |         case 7:
 | 
|---|
 | 57 |             work_unlocked = atoi(argv[5]);
 | 
|---|
 | 58 |         case 6:
 | 
|---|
 | 59 |             work_locked = atoi(argv[5]);
 | 
|---|
| [f7f07f6] | 60 |         case 5:
 | 
|---|
 | 61 |             num_times = atoi(argv[4]);
 | 
|---|
 | 62 |         case 4:
 | 
|---|
 | 63 |             lockCount = atoi(argv[3]);
 | 
|---|
 | 64 |         case 3:
 | 
|---|
 | 65 |             threadCount = atoi(argv[2]);
 | 
|---|
 | 66 |         case 2:
 | 
|---|
 | 67 |             taskCount = atoi(argv[1]);
 | 
|---|
 | 68 |         case 1:
 | 
|---|
 | 69 |             break;
 | 
|---|
 | 70 |         default:
 | 
|---|
| [8f1a99e] | 71 |             break;
 | 
|---|
| [f7f07f6] | 72 |     }
 | 
|---|
 | 73 |         processor p[threadCount];
 | 
|---|
| [cf444b6] | 74 |     linear_backoff_then_block_lock locks[lockCount];
 | 
|---|
| [f7f07f6] | 75 |     worker * worker_arr[taskCount];
 | 
|---|
| [cf444b6] | 76 |     num_times = total_times  / taskCount;
 | 
|---|
| [f7f07f6] | 77 | 
 | 
|---|
| [cf444b6] | 78 |         //printf("Start Test: martin lock simple\n");
 | 
|---|
| [f7f07f6] | 79 |         clock_t begin = clock();
 | 
|---|
 | 80 |         for (unsigned int i = 0; i < taskCount; i++) {
 | 
|---|
| [cf444b6] | 81 |         worker_arr[i] = new( locks, false );
 | 
|---|
| [f7f07f6] | 82 |     }
 | 
|---|
 | 83 |     for (unsigned int i = 0; i < taskCount; i++) {
 | 
|---|
| [cf444b6] | 84 |         delete( worker_arr[i] );
 | 
|---|
| [f7f07f6] | 85 |     }
 | 
|---|
 | 86 |         clock_t end = clock();
 | 
|---|
 | 87 |         double time_spent = (double)(end - begin) / CLOCKS_PER_SEC;
 | 
|---|
| [cf444b6] | 88 |         printf("norm: %f\n", time_spent);
 | 
|---|
 | 89 | 
 | 
|---|
 | 90 |     //printf("Start Test: martin lock improved\n");
 | 
|---|
 | 91 |         begin = clock();
 | 
|---|
 | 92 |         for (unsigned int i = 0; i < taskCount; i++) {
 | 
|---|
 | 93 |         worker_arr[i] = new( locks, true );
 | 
|---|
 | 94 |     }
 | 
|---|
 | 95 |     for (unsigned int i = 0; i < taskCount; i++) {
 | 
|---|
 | 96 |         delete( worker_arr[i] );
 | 
|---|
 | 97 |     }
 | 
|---|
 | 98 |         end = clock();
 | 
|---|
 | 99 |         time_spent = (double)(end - begin) / CLOCKS_PER_SEC;
 | 
|---|
 | 100 |         printf("improved: %f\n", time_spent);
 | 
|---|
| [f7f07f6] | 101 | }
 | 
|---|