From 94ffbb677222e4ab249c2450e67bf0e941fded97 Mon Sep 17 00:00:00 2001
From: Mathieu Faverge <mathieu.faverge@inria.fr>
Date: Tue, 12 Oct 2021 12:18:58 +0200
Subject: [PATCH] starpu: Add kernel information into the tasks
---
runtime/starpu/codelets/codelet_zgemm.c | 8 ++++++++
runtime/starpu/codelets/codelet_zherk.c | 8 ++++++++
runtime/starpu/codelets/codelet_zlauum.c | 8 ++++++++
runtime/starpu/codelets/codelet_zpotrf.c | 8 ++++++++
runtime/starpu/codelets/codelet_zsyrk.c | 8 ++++++++
runtime/starpu/codelets/codelet_ztrmm.c | 8 ++++++++
runtime/starpu/codelets/codelet_ztrsm.c | 8 ++++++++
runtime/starpu/codelets/codelet_ztrtri.c | 8 ++++++++
8 files changed, 64 insertions(+)
diff --git a/runtime/starpu/codelets/codelet_zgemm.c b/runtime/starpu/codelets/codelet_zgemm.c
index e66b85ded..cac587414 100644
--- a/runtime/starpu/codelets/codelet_zgemm.c
+++ b/runtime/starpu/codelets/codelet_zgemm.c
@@ -153,6 +153,14 @@ void INSERT_TASK_zgemm( const RUNTIME_option_t *options,
/* Reduce the C access if needed */
accessC = ( beta == 0. ) ? STARPU_W : STARPU_RW;
+#if defined(CHAMELEON_KERNELS_TRACE)
+ {
+ char *cl_fullname;
+ chameleon_asprintf( &cl_fullname, "%s( %s, %s, %s )", cl_name, clargs->tileA->name, clargs->tileB->name, clargs->tileC->name );
+ cl_name = cl_fullname;
+ }
+#endif
+
/* Insert the task */
rt_starpu_insert_task(
&cl_zgemm,
diff --git a/runtime/starpu/codelets/codelet_zherk.c b/runtime/starpu/codelets/codelet_zherk.c
index 3cc33f423..c4b6eedd8 100644
--- a/runtime/starpu/codelets/codelet_zherk.c
+++ b/runtime/starpu/codelets/codelet_zherk.c
@@ -135,6 +135,14 @@ void INSERT_TASK_zherk( const RUNTIME_option_t *options,
/* Reduce the C access if needed */
accessC = ( beta == 0. ) ? STARPU_W : STARPU_RW;
+#if defined(CHAMELEON_KERNELS_TRACE)
+ {
+ char *cl_fullname;
+ chameleon_asprintf( &cl_fullname, "%s( %s, %s )", cl_name, clargs->tileA->name, clargs->tileC->name );
+ cl_name = cl_fullname;
+ }
+#endif
+
/* Insert the task */
rt_starpu_insert_task(
&cl_zherk,
diff --git a/runtime/starpu/codelets/codelet_zlauum.c b/runtime/starpu/codelets/codelet_zlauum.c
index 700352172..31f801264 100644
--- a/runtime/starpu/codelets/codelet_zlauum.c
+++ b/runtime/starpu/codelets/codelet_zlauum.c
@@ -82,6 +82,14 @@ void INSERT_TASK_zlauum( const RUNTIME_option_t *options,
/* Fix the worker id */
workerid = (schedopt == NULL) ? -1 : schedopt->workerid;
+#if defined(CHAMELEON_KERNELS_TRACE)
+ {
+ char *cl_fullname;
+ chameleon_asprintf( &cl_fullname, "%s( %s )", cl_name, clargs->tileA->name );
+ cl_name = cl_fullname;
+ }
+#endif
+
/* Insert the task */
rt_starpu_insert_task(
&cl_zlauum,
diff --git a/runtime/starpu/codelets/codelet_zpotrf.c b/runtime/starpu/codelets/codelet_zpotrf.c
index 4f58f7239..b0f42d459 100644
--- a/runtime/starpu/codelets/codelet_zpotrf.c
+++ b/runtime/starpu/codelets/codelet_zpotrf.c
@@ -98,6 +98,14 @@ void INSERT_TASK_zpotrf( const RUNTIME_option_t *options,
/* Fix the worker id */
workerid = (schedopt == NULL) ? -1 : schedopt->workerid;
+#if defined(CHAMELEON_KERNELS_TRACE)
+ {
+ char *cl_fullname;
+ chameleon_asprintf( &cl_fullname, "%s( %s )", cl_name, clargs->tileA->name );
+ cl_name = cl_fullname;
+ }
+#endif
+
/* Insert the task */
rt_starpu_insert_task(
&cl_zpotrf,
diff --git a/runtime/starpu/codelets/codelet_zsyrk.c b/runtime/starpu/codelets/codelet_zsyrk.c
index a878aaa19..49a16f71b 100644
--- a/runtime/starpu/codelets/codelet_zsyrk.c
+++ b/runtime/starpu/codelets/codelet_zsyrk.c
@@ -135,6 +135,14 @@ void INSERT_TASK_zsyrk( const RUNTIME_option_t *options,
/* Reduce the C access if needed */
accessC = ( beta == 0. ) ? STARPU_W : STARPU_RW;
+#if defined(CHAMELEON_KERNELS_TRACE)
+ {
+ char *cl_fullname;
+ chameleon_asprintf( &cl_fullname, "%s( %s, %s )", cl_name, clargs->tileA->name, clargs->tileC->name );
+ cl_name = cl_fullname;
+ }
+#endif
+
/* Insert the task */
rt_starpu_insert_task(
&cl_zsyrk,
diff --git a/runtime/starpu/codelets/codelet_ztrmm.c b/runtime/starpu/codelets/codelet_ztrmm.c
index 20f86bd38..887ff71d3 100644
--- a/runtime/starpu/codelets/codelet_ztrmm.c
+++ b/runtime/starpu/codelets/codelet_ztrmm.c
@@ -128,6 +128,14 @@ void INSERT_TASK_ztrmm( const RUNTIME_option_t *options,
/* Fix the worker id */
workerid = (schedopt == NULL) ? -1 : schedopt->workerid;
+#if defined(CHAMELEON_KERNELS_TRACE)
+ {
+ char *cl_fullname;
+ chameleon_asprintf( &cl_fullname, "%s( %s, %s )", cl_name, clargs->tileA->name, clargs->tileB->name );
+ cl_name = cl_fullname;
+ }
+#endif
+
/* Insert the task */
rt_starpu_insert_task(
&cl_ztrmm,
diff --git a/runtime/starpu/codelets/codelet_ztrsm.c b/runtime/starpu/codelets/codelet_ztrsm.c
index e40dde763..2aa43232e 100644
--- a/runtime/starpu/codelets/codelet_ztrsm.c
+++ b/runtime/starpu/codelets/codelet_ztrsm.c
@@ -129,6 +129,14 @@ void INSERT_TASK_ztrsm( const RUNTIME_option_t *options,
/* Fix the worker id */
workerid = (schedopt == NULL) ? -1 : schedopt->workerid;
+#if defined(CHAMELEON_KERNELS_TRACE)
+ {
+ char *cl_fullname;
+ chameleon_asprintf( &cl_fullname, "%s( %s, %s )", cl_name, clargs->tileA->name, clargs->tileB->name );
+ cl_name = cl_fullname;
+ }
+#endif
+
/* Insert the task */
rt_starpu_insert_task(
&cl_ztrsm,
diff --git a/runtime/starpu/codelets/codelet_ztrtri.c b/runtime/starpu/codelets/codelet_ztrtri.c
index 9732e84c3..9605d034e 100644
--- a/runtime/starpu/codelets/codelet_ztrtri.c
+++ b/runtime/starpu/codelets/codelet_ztrtri.c
@@ -96,6 +96,14 @@ void INSERT_TASK_ztrtri( const RUNTIME_option_t *options,
/* Fix the worker id */
workerid = (schedopt == NULL) ? -1 : schedopt->workerid;
+#if defined(CHAMELEON_KERNELS_TRACE)
+ {
+ char *cl_fullname;
+ chameleon_asprintf( &cl_fullname, "%s( %s )", cl_name, clargs->tileA->name );
+ cl_name = cl_fullname;
+ }
+#endif
+
/* Insert the task */
rt_starpu_insert_task(
&cl_ztrtri,
--
GitLab