Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
solverstack
mini-examples
starpu_example_dgemm
Commits
e9048866
Commit
e9048866
authored
Nov 26, 2021
by
Antoine Jego
Browse files
allow 1 node launch
parent
d7111e81
Changes
1
Hide whitespace changes
Inline
Side-by-side
fstarpu_example_dgemm.f90
View file @
e9048866
...
...
@@ -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
7
7
stop
7
9
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
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment