cholesky_compiled.c 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2019-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. /* This is the base code, just like can be read in Chameleon */
  17. /* A source-to-source compiler can very easily produce this kind of
  18. code, with rewritten loops etc */
  19. unsigned k, m, n;
  20. for (k = 0; k < nblocks; k++)
  21. {
  22. POTRF(A(k,k), (2*nblocks - 2*k));
  23. for (m = k+1; m < nblocks; m++)
  24. TRSM(A(k,k), A(m,k), (2*nblocks - 2*k - m));
  25. for (n = k+1; n < nblocks; n++)
  26. {
  27. SYRK(A(n,k), A(n, n), (2*nblocks - 2*k - n));
  28. for (m = n+1; m < nblocks; m++)
  29. GEMM(A(m,k), A(n,k), A(m,n), (2*nblocks - 2*k - n - m));
  30. }
  31. }