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
|
/*
-- MAGMA (version 2.9.0) --
Univ. of Tennessee, Knoxville
Univ. of California, Berkeley
Univ. of Colorado, Denver
@date January 2025
@author Azzam Haidar
@author Raffaele Solca
*/
#ifndef MAGMA_AFFINITY_H
#define MAGMA_AFFINITY_H
#ifndef MAGMA_NOAFFINITY
#ifndef _GNU_SOURCE
#define _GNU_SOURCE
#endif
#include <sched.h>
#if __GLIBC_PREREQ(2,3)
class affinity_set
{
public:
affinity_set();
affinity_set(int cpu);
void add(int cpu);
int get_affinity();
int set_affinity();
void print_affinity(int id, const char* s);
void print_set(int id, const char* s);
private:
cpu_set_t set;
};
#else
#error "Affinity requires Linux glibc version >= 2.3.3, which isn't available. Please add -DMAGMA_NOAFFINITY to the CFLAGS in make.inc."
#endif
#endif // MAGMA_NOAFFINITY
#endif // MAGMA_AFFINITY_H
|
