nf_partition_cl.f90 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. ! StarPU --- Runtime system for heterogeneous multicore architectures.
  2. !
  3. ! Copyright (C) 2016-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. module nf_partition_cl
  17. contains
  18. ! 'cl_partition' codelet routine
  19. recursive subroutine cl_partition_func (buffers, cl_args) bind(C)
  20. use iso_c_binding ! C interfacing module
  21. use fstarpu_mod ! StarPU interfacing module
  22. implicit none
  23. type(c_ptr), value, intent(in) :: buffers, cl_args ! cl_args is unused
  24. real(8), dimension(:,:), pointer :: ma
  25. integer :: ld_ma,nx_ma,ny_ma
  26. integer :: i,j
  27. ld_ma = fstarpu_matrix_get_ld(buffers, 0)
  28. nx_ma = fstarpu_matrix_get_nx(buffers, 0)
  29. ny_ma = fstarpu_matrix_get_ny(buffers, 0)
  30. write(*,*) "ld_ma = ", ld_ma, ", nx_ma = ", nx_ma, ", ny_ma = ", ny_ma
  31. call c_f_pointer(fstarpu_matrix_get_ptr(buffers, 0), ma, shape=[ld_ma,ny_ma])
  32. write(*,*) "ma"
  33. do i=1,nx_ma
  34. do j=1,ny_ma
  35. write(*,*) i,j,ma(i,j)
  36. end do
  37. write(*,*) '-'
  38. end do
  39. end subroutine cl_partition_func
  40. end module nf_partition_cl