|
@@ -21,7 +21,8 @@ program nf_mm
|
|
|
implicit none
|
|
|
|
|
|
logical, parameter :: verbose = .false.
|
|
|
- integer(c_int) :: comm_rank, comm_size, comm_world
|
|
|
+ integer(c_int) :: comm_size, comm_rank
|
|
|
+ integer(c_int), target :: comm_world
|
|
|
integer(c_int) :: N = 16, BS = 4, NB
|
|
|
real(kind=c_double),allocatable,target :: A(:,:), B(:,:), C(:,:)
|
|
|
type(c_ptr),allocatable :: dh_A(:), dh_B(:), dh_C(:,:)
|
|
@@ -164,13 +165,13 @@ program nf_mm
|
|
|
end do
|
|
|
|
|
|
do b_col=1,NB
|
|
|
- do b_row=1,NB
|
|
|
- ret = fstarpu_mpi_task_insert(comm_world, (/ cl_mm, &
|
|
|
- FSTARPU_R, dh_A(b_row), &
|
|
|
- FSTARPU_R, dh_B(b_col), &
|
|
|
- FSTARPU_RW, dh_C(b_row,b_col), &
|
|
|
- C_NULL_PTR /))
|
|
|
- end do
|
|
|
+ do b_row=1,NB
|
|
|
+ call fstarpu_mpi_task_insert((/ c_loc(comm_world), cl_mm, &
|
|
|
+ FSTARPU_R, dh_A(b_row), &
|
|
|
+ FSTARPU_R, dh_B(b_col), &
|
|
|
+ FSTARPU_RW, dh_C(b_row,b_col), &
|
|
|
+ C_NULL_PTR /))
|
|
|
+ end do
|
|
|
end do
|
|
|
|
|
|
call fstarpu_task_wait_for_all()
|