#include "utest_helper.hpp"
#include <cmath>

void compiler_remove_negative_add(void) {
  const int n = 1024;
  float src[n];

  // Setup kernel and buffers
  OCL_CREATE_KERNEL("compiler_remove_negative_add");
  OCL_CREATE_BUFFER(buf[0], 0, n * sizeof(float), NULL);
  OCL_CREATE_BUFFER(buf[1], 0, n * sizeof(float), NULL);
  OCL_SET_ARG(0, sizeof(cl_mem), &buf[0]);
  OCL_SET_ARG(1, sizeof(cl_mem), &buf[1]);
  globals[0] = n;
  locals[0] = 16;

  for (int j = 0; j < 1024; j++) {
    OCL_MAP_BUFFER(0);
    for (int i = 0; i < n; ++i) {
      src[i] = ((float *)buf_data[0])[i] = (j * n + i + 1) * 0.001f;
    }
    OCL_UNMAP_BUFFER(0);

    OCL_NDRANGE(1);

    OCL_MAP_BUFFER(1);
    float *dst = (float *)buf_data[1];
    for (int i = 0; i < n; ++i) {
      float cpu = exp2(-src[i]);
      float gpu = dst[i];
      if (fabsf(cpu - gpu) >= 1e-3) {
        printf("%f %f %f", src[i], cpu, gpu);
        OCL_ASSERT(0);
      }
    }
    OCL_UNMAP_BUFFER(1);
  }
}

MAKE_UTEST_FROM_FUNCTION(compiler_remove_negative_add);