vector_scal_opencl.c 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2010, 2012, 2013 Centre National de la Recherche Scientifique
  4. * Copyright (C) 2010 Institut National de Recherche en Informatique et Automatique
  5. * Copyright (C) 2011 Université de Bordeaux 1
  6. *
  7. * StarPU is free software; you can redistribute it and/or modify
  8. * it under the terms of the GNU Lesser General Public License as published by
  9. * the Free Software Foundation; either version 2.1 of the License, or (at
  10. * your option) any later version.
  11. *
  12. * StarPU is distributed in the hope that it will be useful, but
  13. * WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  15. *
  16. * See the GNU Lesser General Public License in COPYING.LGPL for more details.
  17. */
  18. #include <starpu.h>
  19. extern struct starpu_opencl_program programs;
  20. void scal_opencl_func(void *buffers[], void *_args)
  21. {
  22. float *factor = _args;
  23. int id, devid, err;
  24. cl_kernel kernel;
  25. cl_command_queue queue;
  26. cl_event event;
  27. /* length of the vector */
  28. unsigned n = STARPU_VECTOR_GET_NX(buffers[0]);
  29. /* OpenCL copy of the vector pointer */
  30. cl_mem val = (cl_mem)STARPU_VECTOR_GET_DEV_HANDLE(buffers[0]);
  31. id = starpu_worker_get_id();
  32. devid = starpu_worker_get_devid(id);
  33. err = starpu_opencl_load_kernel(&kernel, &queue, &programs, "vector_mult_opencl",
  34. devid);
  35. if (err != CL_SUCCESS) STARPU_OPENCL_REPORT_ERROR(err);
  36. err = clSetKernelArg(kernel, 0, sizeof(n), &n);
  37. err |= clSetKernelArg(kernel, 1, sizeof(val), &val);
  38. err |= clSetKernelArg(kernel, 2, sizeof(*factor), factor);
  39. if (err) STARPU_OPENCL_REPORT_ERROR(err);
  40. {
  41. size_t global=n;
  42. size_t local;
  43. size_t s;
  44. cl_device_id device;
  45. starpu_opencl_get_device(devid, &device);
  46. err = clGetKernelWorkGroupInfo (kernel, device, CL_KERNEL_WORK_GROUP_SIZE,
  47. sizeof(local), &local, &s);
  48. if (err != CL_SUCCESS) STARPU_OPENCL_REPORT_ERROR(err);
  49. if (local > global) local=global;
  50. err = clEnqueueNDRangeKernel(queue, kernel, 1, NULL, &global, &local, 0,
  51. NULL, &event);
  52. if (err != CL_SUCCESS) STARPU_OPENCL_REPORT_ERROR(err);
  53. }
  54. clFinish(queue);
  55. starpu_opencl_collect_stats(event);
  56. clReleaseEvent(event);
  57. starpu_opencl_release_kernel(kernel);
  58. }