libmcl-sys 0.1.2-rc1

This system crate provides Rust language bindings to the Minos Compute Library (MCL)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142

#include "minos_sched_internal.h"

#include <inttypes.h>
#include <pthread.h>
#include <utlist.h>

#include <atomics.h>

typedef struct lru_list_node {
    enode_t *parent;
    struct lru_list_node *next;
    struct lru_list_node *prev;

    struct lru_list_node *dev_next;
    struct lru_list_node *dev_prev;
} lru_data_t;

pthread_mutex_t lru_lock;
lru_data_t *lru_list;
lru_data_t **dev_lru_list;

void lru_init(mcl_resource_t *res, int ndev) {
    dev_lru_list = malloc(sizeof(lru_data_t *) * ndev);
    memset(dev_lru_list, 0, sizeof(lru_data_t *) * ndev);
    lru_list = NULL;
    pthread_mutex_init(&lru_lock, NULL);
    return;
}

void lru_init_node(enode_t *e) {
    e->pol_data = (void *)malloc(sizeof(lru_data_t));
    memset(e->pol_data, 0, sizeof(lru_data_t));
    ((lru_data_t *)e->pol_data)->parent = e;
}

enode_t *lru_evict(void) {
    pthread_mutex_lock(&lru_lock);
    lru_data_t *el;
    int success = 0;
    DL_FOREACH(lru_list, el) {
        if (el->parent->refs)
            continue;
        success = 1;
        break;
    }

    if (!success) {
        pthread_mutex_unlock(&lru_lock);
        eprintf("Could not find memory not in use");
        return NULL;
    }
    enode_t *ret = el->parent;

    Dprintf("Trying to delete memid %" PRIu64 " from dev %d lru list", ret->mem_data->mem_id, ret->dev);
    DL_DELETE2((dev_lru_list[ret->dev]), el, dev_prev, dev_next);
    DL_DELETE(lru_list, el);

    el->next = NULL;
    el->prev = NULL;
    el->dev_next = NULL;
    el->dev_prev = NULL;

    pthread_mutex_unlock(&lru_lock);
    return ret;
}

enode_t *lru_evict_from_dev(int dev) {
    pthread_mutex_lock(&lru_lock);
    lru_data_t *el;
    int success = 0;
    DL_FOREACH((dev_lru_list[dev]), el) {
        if (el->parent->refs > 0)
            continue;
        success = 1;
        break;
    }
    if (!success) {
        pthread_mutex_unlock(&lru_lock);
        eprintf("Could not find memory not in use");
        return NULL;
    }

    enode_t *ret = el->parent;

    DL_DELETE(lru_list, el);
    DL_DELETE2((dev_lru_list[dev]), el, dev_prev, dev_next);

    el->next = NULL;
    el->prev = NULL;
    el->dev_next = NULL;
    el->dev_prev = NULL;

    pthread_mutex_unlock(&lru_lock);
    return ret;
}

int lru_used(enode_t *e) {
    pthread_mutex_lock(&lru_lock);
    lru_data_t *el = (lru_data_t *)e->pol_data;

    if (el->next || el->prev) {
        DL_DELETE(lru_list, el);
        DL_DELETE2((dev_lru_list[e->dev]), el, dev_prev, dev_next);
    }

    DL_APPEND(lru_list, el);
    DL_APPEND2((dev_lru_list[e->dev]), el, dev_prev, dev_next);
    Dprintf("Added memid %" PRIu64 " to dev %d lru list at %p", e->mem_data->mem_id, e->dev, dev_lru_list[e->dev]);

    ainc(&(e->refs));
    pthread_mutex_unlock(&lru_lock);
    return 0;
}

int lru_released(enode_t *e) {
    adec(&(e->refs));
    return 0;
}

int lru_removed(enode_t *e) {
    pthread_mutex_lock(&lru_lock);
    lru_data_t *el = (lru_data_t *)e->pol_data;
    if (el->next || el->prev) {
        DL_DELETE(lru_list, el);
        DL_DELETE2((dev_lru_list[e->dev]), el, dev_prev, dev_next);
    }
    pthread_mutex_unlock(&lru_lock);
    return 0;
}

void lru_destroy(void) {
}

const struct sched_eviction_policy lru_eviction_policy = {
    .init = lru_init,
    .init_node = lru_init_node,
    .used = lru_used,
    .released = lru_released,
    .removed = lru_removed,
    .destroy = lru_destroy,
    .evict = lru_evict,
    .evict_from_dev = lru_evict_from_dev};