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
|
/*
* Support file for amdgcn in newlib.
* Copyright (c) 2014, 2017 Mentor Graphics.
*
* The authors hereby grant permission to use, copy, modify, distribute,
* and license this software and its documentation for any purpose, provided
* that existing copyright notices are retained in all copies and that this
* notice is included verbatim in any distributions. No written agreement,
* license, or royalty fee is required for any of the authorized uses.
* Modifications to this software may be copyrighted by their authors
* and need not follow the licensing terms described here, provided that
* the new terms are clearly indicated on the first page of each file where
* they apply.
*/
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
/* This struct must match the one used by gcn-run and libgomp.
It holds all the data output from a kernel (besides mapping data).
The base address pointer can be found at kernargs+16.
The next_output counter must be atomically incremented for each
print output. Only when the print data is fully written can the
"written" flag be set.
The buffer is circular; the host increments the consumed counter
and clears the written flag as it goes, opening up slots for reuse.
The counters always use absolute numbers. */
struct output {
int return_value;
unsigned int next_output;
struct printf_data {
int written;
char msg[128];
int type;
union {
int64_t ivalue;
double dvalue;
char text[128];
};
} queue[1024];
unsigned int consumed;
};
_READ_WRITE_RETURN_TYPE write (int fd, const void *buf, size_t count)
{
if (fd != 1 && fd != 2)
{
errno = EBADF;
return -1;
}
/* The output data is at ((void*)kernargs)[2]. */
#if defined(__has_builtin) && __has_builtin(__builtin_gcn_kernarg_ptr)
register void **kernargs = __builtin_gcn_kernarg_ptr ();
#else
register void **kernargs asm("s8");
#endif
struct output *data = (struct output *)kernargs[2];
/* Each output slot allows 256 bytes, so reserve as many as we need. */
unsigned int slot_count = ((count+1)/256)+1;
unsigned int index = __atomic_fetch_add (&data->next_output, slot_count,
__ATOMIC_ACQUIRE);
if ((unsigned int)(index + slot_count) < data->consumed)
{
/* Overflow. */
errno = EFBIG;
return 0;
}
for (int c = count;
c >= 0;
buf += 256, c -= 256, index++)
{
unsigned int slot = index % 1024;
/* Spinlock while the host catches up. */
if (index >= 1024)
while (__atomic_load_n (&data->consumed, __ATOMIC_ACQUIRE)
<= (index - 1024))
asm ("s_sleep 64");
if (c < 128)
{
memcpy (data->queue[slot].msg, buf, c);
data->queue[slot].msg[c] = '\0';
data->queue[slot].text[0] = '\0';
}
else if (c < 256)
{
memcpy (data->queue[slot].msg, buf, 128);
memcpy (data->queue[slot].text, buf+128, c-128);
data->queue[slot].text[c-128] = '\0';
}
else
{
memcpy (data->queue[slot].msg, buf, 128);
memcpy (data->queue[slot].text, buf+128, 128);
}
data->queue[slot].type = 3; /* Raw. */
__atomic_store_n (&data->queue[slot].written, 1, __ATOMIC_RELEASE);
}
return count;
}
|
