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
|
#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <sys/mman.h>
#include <error.h>
#include <errno.h>
#include "NE10.h"
/* customize these macros and variables for a particular implementation */
#define FFT_IN_ROUNDS (1)
#define FFT_CALC_POINTS (256)
#define FFT_CALC_ROUNDS (16)
#define FFT_POINTS (FFT_CALC_POINTS * FFT_CALC_ROUNDS)
#define IN_SAMPLE_TYPE ne10_fft_cpx_int32_t
#define OUT_SAMPLE_TYPE ne10_fft_cpx_int32_t
#define CFG_TYPE ne10_fft_cfg_int32_t
#define CFG_ALLOC_FUNC ne10_fft_alloc_c2c_int32
#define FFT_FUNC ne10_fft_c2c_1d_int32_neon
const char *g_test = "stream_neon32_256x16x1";
const char *g_description = "256x16x1 point FFT, NEON implementation, 32-bit precision";
/* these functions and variables are implemented in overhead.c as a library */
extern void initialize_everything(int argc, char **argv);
extern uint64_t get_gt_value(void);
extern void release_everything(void);
extern void print_results(void);
extern char *g_input_filename;
extern char *g_output_filename;
extern uint64_t g_start_time;
extern uint64_t g_end_time;
int main(int argc, char **argv)
{
int input_fd;
int output_fd;
int raw256stream_dev_fd;
int result;
IN_SAMPLE_TYPE *in_buf;
OUT_SAMPLE_TYPE *out_buf;
CFG_TYPE cfg;
int i;
int j;
initialize_everything(argc, argv);
/* allocate storage for input, output and config buffers */
in_buf = (IN_SAMPLE_TYPE*) NE10_MALLOC (FFT_POINTS * sizeof(IN_SAMPLE_TYPE));
if(in_buf == NULL)
error(1, errno, "in_buf allocation");
out_buf = (OUT_SAMPLE_TYPE*) NE10_MALLOC (FFT_POINTS * sizeof(OUT_SAMPLE_TYPE));
if(out_buf == NULL)
error(1, errno, "out_buf allocation");
cfg = CFG_ALLOC_FUNC(FFT_CALC_POINTS);
if(cfg == NULL)
error(1, errno, "cfg allocation");
/* open the input and output files and raw256stream device */
input_fd = open(g_input_filename, O_RDONLY);
if(input_fd < 0)
error(1, errno, "opening input file '%s'", g_input_filename);
output_fd = open(g_output_filename, O_WRONLY | O_CREAT, ~((mode_t)umask));
if(output_fd < 0)
error(1, errno, "opening output file '%s'", g_output_filename);
raw256stream_dev_fd = open("/dev/raw256stream", O_RDWR);
if(raw256stream_dev_fd < 0)
error(1, errno, "opening raw256stream_dev_fd");
/* read the input data */
result = read(input_fd, in_buf, FFT_CALC_POINTS * sizeof(IN_SAMPLE_TYPE));
if(result < 0)
error(1, errno, "read input file");
if(result != (FFT_CALC_POINTS * sizeof(IN_SAMPLE_TYPE)))
error(1, 0, "input data size, expected %d but got %d", FFT_CALC_POINTS * sizeof(IN_SAMPLE_TYPE), result);
/* write the waveform buffer */
result = write(raw256stream_dev_fd, in_buf, FFT_CALC_POINTS * sizeof(IN_SAMPLE_TYPE));
if(result < 0)
error(1, errno, "write waveform buffer");
if(result != (FFT_CALC_POINTS * sizeof(IN_SAMPLE_TYPE)))
error(1, 0, "output data size, expected %d but got %d", FFT_CALC_POINTS * sizeof(IN_SAMPLE_TYPE), result);
/* capture the start value of the GT */
g_start_time = get_gt_value();
for(j = 0 ; j < FFT_IN_ROUNDS ; j++) {
/* read the input stream */
result = read(raw256stream_dev_fd, in_buf, FFT_POINTS * sizeof(IN_SAMPLE_TYPE));
if(result < 0)
error(1, errno, "read input stream");
if(result != (FFT_POINTS * sizeof(IN_SAMPLE_TYPE)))
error(1, 0, "input data size, expected %d but got %d", FFT_POINTS * sizeof(IN_SAMPLE_TYPE), result);
/* compute FFT */
for (i = 0; i < FFT_CALC_ROUNDS ; i++) {
FFT_FUNC(out_buf + (i * FFT_CALC_POINTS), in_buf + (i * FFT_CALC_POINTS), cfg, 0, 1);
}
/* write the output data */
result = write(output_fd, out_buf, FFT_POINTS * sizeof(OUT_SAMPLE_TYPE));
if(result < 0)
error(1, errno, "write output file");
if(result != (FFT_POINTS * sizeof(OUT_SAMPLE_TYPE)))
error(1, 0, "output data size, expected %d but got %d", FFT_POINTS * sizeof(OUT_SAMPLE_TYPE), result);
}
/* capture the end value of the GT */
g_end_time = get_gt_value();
/* close the input and output files and stream device */
close(raw256stream_dev_fd);
close(output_fd);
close(input_fd);
/* free storage for input, output and config buffers */
NE10_FREE (cfg);
NE10_FREE (out_buf);
NE10_FREE (in_buf);
print_results();
release_everything();
return 0;
}
|
