diff options
Diffstat (limited to 'final/Bitcode/Regression/fft/driver.cpp')
-rw-r--r-- | final/Bitcode/Regression/fft/driver.cpp | 65 |
1 files changed, 65 insertions, 0 deletions
diff --git a/final/Bitcode/Regression/fft/driver.cpp b/final/Bitcode/Regression/fft/driver.cpp new file mode 100644 index 00000000..ea2bfd21 --- /dev/null +++ b/final/Bitcode/Regression/fft/driver.cpp @@ -0,0 +1,65 @@ +#include "halide_buffer.h" + +template<typename T> +T rand_value() { + return (T)(rand() * 0.125) - 100; +} + +template<typename T> +buffer_t make_buffer(int w, int h) { + T *mem = new T[w*h]; + buffer_t buf = {0}; + buf.host = (uint8_t *)mem; + buf.extent[0] = w; + buf.extent[1] = h; + buf.elem_size = sizeof(T); + buf.stride[0] = 1; + buf.stride[1] = w; + + for (int i = 0; i < w*h; i++) { + mem[i] = rand_value<T>(); + } + + return buf; +} + + +int main(int argc, char **argv) { + unsigned int err_code = 0; +#if (!__has_builtin(__builtin_cpu_supports) && (defined(__i386__) || defined(__x86_64__))) + return err_code; +#endif +#if defined(__i386__) || defined(__x86_64__) + if (!__builtin_cpu_supports("avx")) { + return err_code; + } +#endif + time_t seed; + if (argc > 1) { + seed = atoi(argv[1]); + } + else { + seed = time(NULL); + srand (seed); + } + const int W = 256, H = 256; + // Make some input buffers + buffer_t bufs[] = { + make_buffer<int32_t>(W, H), + make_buffer<int32_t>(W, H), + make_buffer<int32_t>(W, H) + }; + + for (int i = 0; i < 24; i++) { + filter f = filters[i]; + f.fn(bufs + 0, + bufs + 1, + bufs + 2); + } + + for (int i = 0; i < sizeof(bufs)/sizeof(buffer_t); i++) { + delete[] bufs[i].host; + } + + return err_code; +} |