rocprim/trunk/test.cpp

   1 #include <rocprim/rocprim.hpp>
   2 #include <vector>
   3 #include <iostream>
   4 #include <random>
   5 #include <algorithm>
   6 #include <cmath>
   7
   8 int main()
   9 {
  10     auto xpy = [] __device__(float x, float y) -> float{
  11         return x + y;
  12     };
  13
  14     size_t size = 1024;
  15     std::vector<float> xin(size);
  16     std::vector<float> yin(size);
  17
  18     std::random_device rd;
  19     std::mt19937 gen(rd());
  20     std::uniform_real_distribution<float> dist(-1.0, 1.0);
  21
  22     auto myrand = [&]() -> float {return dist(gen);};
  23
  24     std::generate(xin.begin(), xin.end(), myrand);
  25     std::generate(yin.begin(), yin.end(), myrand);
  26
  27     std::vector<float> zref(size);
  28     for(size_t i = 0; i < size; i++){
  29         zref[i] = xin[i] + yin[i];
  30     }
  31
  32     float *x;
  33     float *y;
  34     float *z;
  35     hipMalloc((void**)&x, sizeof *x * size);
  36     hipMalloc((void**)&y, sizeof *y * size);
  37     hipMalloc((void**)&z, sizeof *z * size);
  38
  39     hipMemcpy(x, xin.data(), sizeof *x * size, hipMemcpyHostToDevice);
  40     hipMemcpy(y, yin.data(), sizeof *y * size, hipMemcpyHostToDevice);
  41
  42     rocprim::transform(x, y, z, size, xpy);
  43
  44     std::vector<float> zout(size);
  45     hipMemcpy(zout.data(), z, sizeof *z * size, hipMemcpyDeviceToHost);
  46
  47     for(size_t i = 0; i < size; i++){
  48         if(std::abs(zout[i] - zref[i]) > 0.001f){
  49             std::cout << "Element mismatch at index " << i << "\n";
  50             std::cout << "Got " << zout[i] << " but expected " << zref[i] << "\n";
  51             return 1;
  52         }
  53     }
  54     std::cout << "TESTS PASSED!" << std::endl;
  55
  56     hipFree(x);
  57     hipFree(y);
  58     hipFree(z);
  59 }