blob: 4564cae3f260b05a7cd4e3d6637eb6f9c2d34b3c (
plain)
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
|
/* Copyright (C) 2012-2016 Free Software Foundation, Inc.
This file is part of the GNU C Library.
The GNU C Library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.
The GNU C Library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public
License along with the GNU C Library. If not, see
<http://www.gnu.org/licenses/>. */
#include <errno.h>
#include <stddef.h>
#include <fcntl.h>
#include <unistd.h>
#include <sched.h>
#include <sysdep.h>
#include <arch/spr_def.h>
/* The count of cores horizontally (X dimension) on the chip. */
static int chip_width;
/* Read the chip "width" from the /sys filesystem. */
static int
initialize_chip_width (void)
{
int w = 0;
int fd;
fd = __open ("/sys/devices/system/cpu/chip_width", O_RDONLY);
if (fd >= 0)
{
char buf[64];
ssize_t n;
int i;
n = __read (fd, buf, sizeof (buf));
__close (fd);
for (i = 0; i < n; ++i)
{
if (buf[i] < '0' || buf[i] > '9')
break;
w = (w * 10) + (buf[i] - '0');
}
}
/* Store a negative value so we don't try again. */
if (w == 0)
w = -1;
/* Using an atomic idempotent write here makes this thread-safe. */
chip_width = w;
return w;
}
int
sched_getcpu (void)
{
unsigned int coord;
int w = chip_width;
if (__builtin_expect (w <= 0, 0))
{
if (w == 0)
w = initialize_chip_width ();
if (w < 0)
{
unsigned int cpu;
int r = INLINE_SYSCALL (getcpu, 3, &cpu, NULL, NULL);
return r == -1 ? r : cpu;
}
}
/* Assign 64-bit value to a 32-bit variable to ensure 32-bit multiply. */
coord = __insn_mfspr (SPR_TILE_COORD);
/* Extract Y coord from bits 7..10 and X coord from bits 18..21. */
return ((coord >> 7) & 0xf) * w + ((coord >> 18) & 0xf);
}
|
