Commit e9048866 authored by Antoine Jego's avatar Antoine Jego
Browse files

allow 1 node launch

parent d7111e81
......@@ -66,26 +66,21 @@ program fstarpu_example_dgemm
ret = fstarpu_mpi_init(1)
if (ret /= 0) then
write(*,'("fstarpu_mpi_init status:",i4)') ret
stop 1
stop 2
end if
! stop there if no CPU worker available
ncpu = fstarpu_cpu_worker_get_count()
if (ncpu == 0) then
write(*,'("no cpu workers available")')
call fstarpu_shutdown()
stop 77
stop 79
end if
comm_world = fstarpu_mpi_world_comm()
comm_size = fstarpu_mpi_world_size()
comm_rank = fstarpu_mpi_world_rank()
if (comm_size < 2) then
call fstarpu_shutdown()
ret = fstarpu_mpi_shutdown()
stop 77
end if
call get_command_argument(1, value=str, length=i)
read(str(1:i),*) m
call get_command_argument(2, value=str, length=i)
......@@ -117,6 +112,7 @@ program fstarpu_example_dgemm
if (mod(m,bs).ne.0) stop 75
if (mod(n,bs).ne.0) stop 75
if (mod(k,bs).ne.0) stop 75
if (p*q.ne.comm_size) stop 74
mb = m/bs
nb = n/bs
kb = k/bs
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment