matvecmult_kernel.cl 920 B

1234567891011121314151617181920212223242526272829303132
  1. /*
  2. * StarPU --- Runtime system for heterogeneous multicore architectures.
  3. *
  4. * Copyright (C) 2014 Université de Bordeaux
  5. *
  6. * StarPU is free software; you can redistribute it and/or modify
  7. * it under the terms of the GNU Lesser General Public License as published by
  8. * the Free Software Foundation; either version 2.1 of the License, or (at
  9. * your option) any later version.
  10. *
  11. * StarPU is distributed in the hope that it will be useful, but
  12. * WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  14. *
  15. * See the GNU Lesser General Public License in COPYING.LGPL for more details.
  16. */
  17. __kernel void matVecMult(const __global float *A, const __global float *X, int n, int m, __global float *Y)
  18. {
  19. const int i = get_global_id(0);
  20. if (i < m)
  21. {
  22. float val = 0;
  23. int j;
  24. for (j = 0; j < n; j++)
  25. val += A[i*n+j] * X[j];
  26. Y[i] = val;
  27. }
  28. }