From 7ca196ea186d1f725e3ed9ec624e5681beaf13c3 Mon Sep 17 00:00:00 2001
From: Florent Pruvost <florent.pruvost@inria.fr>
Date: Tue, 28 Jul 2015 11:54:36 +0000
Subject: [PATCH] force execution on the node owning A2, correct some bugs in
StarPU when several data are in W mode
---
runtime/starpu/codelets/codelet_ztsmqr.c | 1 +
runtime/starpu/codelets/codelet_ztsqrt.c | 1 +
2 files changed, 2 insertions(+)
diff --git a/runtime/starpu/codelets/codelet_ztsmqr.c b/runtime/starpu/codelets/codelet_ztsmqr.c
index 554688432..282e58d72 100644
--- a/runtime/starpu/codelets/codelet_ztsmqr.c
+++ b/runtime/starpu/codelets/codelet_ztsmqr.c
@@ -173,6 +173,7 @@ void MORSE_TASK_ztsmqr(MORSE_option_t *options,
STARPU_VALUE, &ldwork, sizeof(int),
STARPU_PRIORITY, options->priority,
STARPU_CALLBACK, callback,
+ STARPU_EXECUTE_ON_NODE, A2->get_rankof(A2, A2m, A2n),
0);
}
}
diff --git a/runtime/starpu/codelets/codelet_ztsqrt.c b/runtime/starpu/codelets/codelet_ztsqrt.c
index 84b37b64b..240605594 100644
--- a/runtime/starpu/codelets/codelet_ztsqrt.c
+++ b/runtime/starpu/codelets/codelet_ztsqrt.c
@@ -129,6 +129,7 @@ void MORSE_TASK_ztsqrt(MORSE_option_t *options,
STARPU_VALUE, &h_work, sizeof(MORSE_starpu_ws_t *),
STARPU_PRIORITY, options->priority,
STARPU_CALLBACK, callback,
+ STARPU_EXECUTE_ON_NODE, A2->get_rankof(A2, A2m, A2n),
0);
}
}
--
GitLab