dot_product_opencl_kernels.cl 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2012 inria
  4. *
  5. * StarPU is free software; you can redistribute it and/or modify
  6. * it under the terms of the GNU Lesser General Public License as published by
  7. * the Free Software Foundation; either version 2.1 of the License, or (at
  8. * your option) any later version.
  9. *
  10. * StarPU is distributed in the hope that it will be useful, but
  11. * WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  13. *
  14. * See the GNU Lesser General Public License in COPYING.LGPL for more details.
  15. */
  16. #include "dot_product.h"
  17. #pragma OPENCL EXTENSION cl_khr_fp64 : enable
  18. __kernel void _redux_opencl(__global DOT_TYPE *dota,
  19. __global DOT_TYPE *dotb)
  20. {
  21. const int i = get_global_id(0);
  22. *dota += *dotb;
  23. }
  24. __kernel void _dot_opencl(__global float *x,
  25. __global float *y,
  26. __global DOT_TYPE *dot,
  27. unsigned n)
  28. {
  29. unsigned i;
  30. __local double tmp;
  31. tmp = 0.0;
  32. for (i = 0; i < n ; i++)
  33. tmp += x[i]*y[i];
  34. *dot += tmp;
  35. }