Update zgeqrf to the new recursive API in parsec

Signed-off-by: Aurélien Bouteiller <bouteill@icl.utk.edu>
ICLDisco · Sep 5, 2023 · 39da035 · 39da035
1 parent ef64b72
commit 39da035
Showing 1 changed file with 27 additions and 4 deletions.
diff --git a/src/zgeqrf.jdf b/src/zgeqrf.jdf
@@ -15,6 +15,10 @@ extern "C" %{
 
 #include "parsec/data_dist/matrix/subtile.h"
 #include "parsec/recursive.h"
+static void zgeqrt_recursive_cb(parsec_taskpool_t* tp, const parsec_recursive_callback_t* data);
+static void zunmqr_recursive_cb(parsec_taskpool_t* tp, const parsec_recursive_callback_t* data);
+static void ztsqrt_recursive_cb(parsec_taskpool_t* tp, const parsec_recursive_callback_t* data);
+static void ztsmqr_recursive_cb(parsec_taskpool_t* tp, const parsec_recursive_callback_t* data);
 
 #if defined(DPLASMA_HAVE_CUDA)
 #include "cores/dplasma_zcores.h"
@@ -150,7 +154,7 @@ BODY [type=RECURSIVE]
 
         /* recursive call */
         parsec_recursivecall((parsec_task_t*)this_task,
-                             parsec_zgeqrt, dplasma_zgeqrfr_geqrt_Destruct,
+                             parsec_zgeqrt, zgeqrt_recursive_cb,
                              2, small_descA, small_descT);
 
         return PARSEC_HOOK_RETURN_ASYNC;
@@ -246,7 +250,7 @@ BODY [type=RECURSIVE]
 
         /* recursive call */
         parsec_recursivecall((parsec_task_t*)this_task,
-                             parsec_zunmqr_panel, dplasma_zgeqrfr_unmqr_Destruct,
+                             parsec_zunmqr_panel, zunmqr_recursive_cb,
                              3, small_descA, small_descC, small_descT);
 
         return PARSEC_HOOK_RETURN_ASYNC;
@@ -374,7 +378,7 @@ BODY [type=RECURSIVE]
 
         /* recursive call */
         parsec_recursivecall((parsec_task_t*)this_task,
-                             parsec_ztsqrt, dplasma_zgeqrfr_tsqrt_Destruct,
+                             parsec_ztsqrt, ztsqrt_recursive_cb,
                              3, small_descA1, small_descA2, small_descT);
 
         return PARSEC_HOOK_RETURN_ASYNC;
@@ -542,7 +546,7 @@ BODY [type=RECURSIVE]
 
         /* recursive call */
         parsec_recursivecall((parsec_task_t*)this_task,
-                             parsec_ztsmqr, dplasma_zgeqrfr_tsmqr_Destruct,
+                             parsec_ztsmqr, ztsmqr_recursive_cb,
                              4, small_descA1, small_descA2, small_descV, small_descT);
 
         return PARSEC_HOOK_RETURN_ASYNC;
@@ -580,3 +584,22 @@ BODY
 
 }
 END
+
+extern "C" %{
+static void zgeqrt_recursive_cb(parsec_taskpool_t* tp, const parsec_recursive_callback_t* data) {
+    (void)data;
+    dplasma_zgeqrfr_geqrt_Destruct(tp);
+}
+static void zunmqr_recursive_cb(parsec_taskpool_t* tp, const parsec_recursive_callback_t* data) {
+    (void)data;
+    dplasma_zgeqrfr_unmqr_Destruct(tp);
+}
+static void ztsqrt_recursive_cb(parsec_taskpool_t* tp, const parsec_recursive_callback_t* data) {
+    (void)data;
+    dplasma_zgeqrfr_tsqrt_Destruct(tp);
+}
+static void ztsmqr_recursive_cb(parsec_taskpool_t* tp, const parsec_recursive_callback_t* data) {
+    (void)data;
+    dplasma_zgeqrfr_tsmqr_Destruct(tp);
+}
+%}