nf_partition_cl.f90 1.6 KB

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