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
9b4a1fcd
Commit
9b4a1fcd
authored
Dec 06, 2021
by
Antoine Jego
Browse files
actually use context
parent
0f26a756
Changes
1
Hide whitespace changes
Inline
Side-by-side
starpu_example_dgemm.c
View file @
9b4a1fcd
...
...
@@ -658,11 +658,20 @@ int main(int argc, char *argv[])
b_row
,
b_aisle
,
Ail
->
registered
,
Ail
->
hdl
,
Ail
->
owner
,
b_aisle
,
b_col
,
Blj
->
registered
,
Blj
->
hdl
,
Blj
->
owner
);
}
starpu_mpi_task_insert
(
MPI_COMM_WORLD
,
&
gemm_cl
,
STARPU_CL_ARGS
,
clargs
,
sizeof
(
struct
cl_zgemm_args_s
),
STARPU_R
,
Ail
->
hdl
,
STARPU_R
,
Blj
->
hdl
,
STARPU_RW
,
Cij
->
hdl
,
0
);
if
(
context
)
{
starpu_mpi_task_insert
(
MPI_COMM_WORLD
,
&
gemm_cl
,
STARPU_CL_ARGS
,
clargs
,
sizeof
(
struct
cl_zgemm_args_s
),
STARPU_R
,
Ail
->
hdl
,
STARPU_R
,
Blj
->
hdl
,
STARPU_RW
,
Cij
->
hdl
,
0
);
}
else
{
starpu_mpi_task_insert
(
MPI_COMM_WORLD
,
&
gemm_cl
,
STARPU_CL_ARGS
,
clargs
,
sizeof
(
struct
cl_zgemm_args_s
),
STARPU_SCHED_CTX
,
ctx
,
STARPU_R
,
Ail
->
hdl
,
STARPU_R
,
Blj
->
hdl
,
STARPU_RW
,
Cij
->
hdl
,
0
);
}
}
else
{
// printf("[%d] NOT inserted C_%d,%d += A_%d,%d B_%d,%d\n",comm_rank, b_row,b_col, b_row,b_aisle, b_aisle,b_col);
}
...
...
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