From e93ae5f59cc43f63edac39898337f4438b33d941 Mon Sep 17 00:00:00 2001
From: Mathieu Faverge <mathieu.faverge@inria.fr>
Date: Tue, 9 Jan 2024 11:55:12 +0100
Subject: [PATCH] starpu/lacpy: make sure both src and dst tiles have the same
size when using the starpu copy mechanism
---
runtime/starpu/codelets/codelet_zlacpy.c | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/runtime/starpu/codelets/codelet_zlacpy.c b/runtime/starpu/codelets/codelet_zlacpy.c
index bf0466b75..40bbca3a2 100644
--- a/runtime/starpu/codelets/codelet_zlacpy.c
+++ b/runtime/starpu/codelets/codelet_zlacpy.c
@@ -164,6 +164,7 @@ void INSERT_TASK_zlacpyx( const RUNTIME_option_t *options,
/* Insert the task */
if ( (uplo == ChamUpperLower) &&
(tileA->m == m) && (tileA->n == n) &&
+ (tileB->m == m) && (tileB->n == n) &&
(displA == 0) && (displB == 0) )
{
#if defined(CHAMELEON_USE_MPI)
@@ -238,7 +239,8 @@ void INSERT_TASK_zlacpy( const RUNTIME_option_t *options,
#if !defined(CHAMELEON_USE_MPI) || defined(HAVE_STARPU_MPI_DATA_CPY_PRIORITY)
/* Insert the task */
if ( (uplo == ChamUpperLower) &&
- (tileA->m == m) && (tileA->n == n) )
+ (tileA->m == m) && (tileA->n == n) &&
+ (tileB->m == m) && (tileB->n == n) )
{
#if defined(CHAMELEON_USE_MPI)
insert_task_zlacpy_on_remote_node( options,
--
GitLab