nf_dynbuf.f90 2.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. program nf_dynbuf
  2. use iso_c_binding ! C interfacing module
  3. use fstarpu_mod ! StarPU interfacing module
  4. use nf_dynbuf_cl
  5. implicit none
  6. type(c_ptr) :: cl_dynbuf_big ! a pointer for the codelet structure
  7. type(c_ptr) :: dh_var
  8. type(c_ptr) :: descrs_var
  9. integer(c_int),target :: nbuffers
  10. integer(c_int) :: err ! return status for fstarpu_init
  11. integer(c_int) :: ncpu ! number of cpus workers
  12. integer(c_int),target :: var
  13. integer(c_int) :: i
  14. var = 42
  15. ! initialize StarPU with default settings
  16. err = fstarpu_init(C_NULL_PTR)
  17. if (err == -19) then
  18. stop 77
  19. end if
  20. ! stop there if no CPU worker available
  21. ncpu = fstarpu_cpu_worker_get_count()
  22. if (ncpu == 0) then
  23. call fstarpu_shutdown()
  24. stop 77
  25. end if
  26. ! allocate an empty codelet structure
  27. cl_dynbuf_big = fstarpu_codelet_allocate()
  28. call fstarpu_codelet_set_name(cl_dynbuf_big, C_CHAR_"dummy_big_kernel"//C_NULL_CHAR)
  29. call fstarpu_codelet_add_cpu_func(cl_dynbuf_big, C_FUNLOC(cl_cpu_func_dynbuf_big))
  30. write(*,*) "FSTARPU_NMAXBUFS",FSTARPU_NMAXBUFS
  31. nbuffers = FSTARPU_NMAXBUFS+1
  32. call fstarpu_codelet_set_nbuffers(cl_dynbuf_big, nbuffers)
  33. call fstarpu_variable_data_register(dh_var, 0, c_loc(var), c_sizeof(var))
  34. descrs_var = fstarpu_data_descr_array_alloc(nbuffers)
  35. do i=0,nbuffers-1
  36. call fstarpu_data_descr_array_set(descrs_var, i, dh_var, FSTARPU_RW)
  37. end do
  38. call fstarpu_insert_task((/ cl_dynbuf_big, &
  39. FSTARPU_VALUE, c_loc(nbuffers), FSTARPU_SZ_C_INT, &
  40. FSTARPU_DATA_MODE_ARRAY, descrs_var, c_loc(nbuffers), &
  41. C_NULL_PTR /))
  42. call fstarpu_task_wait_for_all()
  43. call fstarpu_data_descr_array_free(descrs_var)
  44. call fstarpu_data_unregister(dh_var)
  45. ! free codelet structure
  46. call fstarpu_codelet_free(cl_dynbuf_big)
  47. ! shut StarPU down
  48. call fstarpu_shutdown()
  49. end program nf_dynbuf