#include <math.h>
#include <stdio.h>
#include <cstdlib>
#include <string>
static void fn() {
}
int main(int argc, char** argv) {
double peak = 0;
try {
int device = argc > 1 ? atoi(argv[1]) : 0;
const std::string
dtype(argc > 2 ? argv[2] :
"f32");
printf("Device %d isHalfAvailable ? %s\n", device,
printf(
"Benchmark N-by-N matrix multiply at %s \n",
dtype.c_str());
for (int n = 128; n <= 2048; n += 128) {
printf("%4d x %4d: ", n, n);
double gflops = 2.0 * powf(n, 3) / (time * 1e9);
if (gflops > peak) peak = gflops;
printf(" %4.0f Gflops\n", gflops);
fflush(stdout);
}
fprintf(stderr,
"%s\n", e.
what());
throw;
}
printf(" ### peak %g GFLOPS\n", peak);
return 0;
}
A multi dimensional data container.
An ArrayFire exception class.
virtual const char * what() const
Returns an error message for the exception in a string format.
@ f32
32-bit floating point values
@ f16
16-bit floating point value
array matmul(const array &lhs, const array &rhs, const matProp optLhs=AF_MAT_NONE, const matProp optRhs=AF_MAT_NONE)
C++ Interface to multiply two matrices.
array constant(T val, const dim4 &dims, const dtype ty=(af_dtype) dtype_traits< T >::ctype)
C++ Interface to generate an array with elements set to a specified value.
bool isHalfAvailable(const int device)
Queries the current device for half precision floating point support.
void setDevice(const int device)
Sets the current device.
double timeit(void(*fn)())