#include <stdio.h>
#include <math.h>
#include <cstdlib>
static void fn()
{
}
int main(int argc, char ** argv)
{
try {
int device = argc > 1 ? atoi(argv[1]) : 0;
printf("Benchmark N-by-N 2D fft\n");
for (int M = 7; M <= 12; M++) {
int N = (1 << M);
printf("%4d x %4d: ", N, N);
double gflops = 10.0 * N * N * M / (time * 1e9);
printf(" %4.0f Gflops\n", gflops);
fflush(stdout);
}
fprintf(stderr,
"%s\n", e.
what());
}
return 0;
}
A multi dimensional data container.
Definition: array.h:27
Definition: exception.h:20
virtual const char * what() const
Definition: exception.h:34
AFAPI array randu(const dim4 &dims, const dtype ty=f32)
AFAPI void setDevice(const int device)
Sets the current device.
void eval() const
Evaluate any JIT expressions to generate data for the array.
AFAPI array fft2(const array &in, const dim_t odim0=0, const dim_t odim1=0)
C++ Interface for fast fourier transform on two dimensional signals.
Definition: algorithm.h:15
AFAPI double timeit(void(*fn)())