matvecmult_kernel.cl 961 B

12345678910111213141516171819202122232425262728293031
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2010-2021 Université de Bordeaux, CNRS (LaBRI UMR 5800), 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. __kernel void matVecMult(const __global float *A, const __global float *X, int n, int m, __global float *Y, int ld)
  17. {
  18. const int i = get_global_id(0);
  19. if (i < m)
  20. {
  21. float val = 0;
  22. int j;
  23. for (j = 0; j < n; j++)
  24. val += A[i*ld+j] * X[j];
  25. Y[i] = val;
  26. }
  27. }