cholesky_codelets.jl 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. # StarPU --- Runtime system for heterogeneous multicore architectures.
  2. #
  3. # Copyright (C) 2020 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. chol_model11 = starpu_perfmodel(
  17. perf_type = starpu_perfmodel_type(STARPU_HISTORY_BASED),
  18. symbol = "chol_model11"
  19. )
  20. chol_model21 = starpu_perfmodel(
  21. perf_type = starpu_perfmodel_type(STARPU_HISTORY_BASED),
  22. symbol = "chol_model21"
  23. )
  24. chol_model22 = starpu_perfmodel(
  25. perf_type = starpu_perfmodel_type(STARPU_HISTORY_BASED),
  26. symbol = "chol_model22"
  27. )
  28. cl_11 = starpu_codelet(
  29. cpu_func = "u11",
  30. cuda_func = "u11",
  31. modes = [STARPU_RW],
  32. color = 0xffff00,
  33. perfmodel = chol_model11
  34. )
  35. cl_21 = starpu_codelet(
  36. cpu_func = "u21",
  37. cuda_func = "u21",
  38. modes = [STARPU_R, STARPU_RW],
  39. color = 0x8080ff,
  40. perfmodel = chol_model21
  41. )
  42. cl_22 = starpu_codelet(
  43. cpu_func = "u22",
  44. cuda_func = "u22",
  45. modes = [STARPU_R, STARPU_R, STARPU_RW],
  46. color = 0x00ff00,
  47. perfmodel = chol_model22
  48. )