starpu/codelets/convert: add a conversion codelet to automatically change the precision on request

runtime/starpu/CMakeLists.txt

@@ -250,6 +250,11 @@ precisions_rules_py(RUNTIME_SRCS_GENERATED "${ZSRC}"
   PRECISIONS "${CHAMELEON_PRECISION}"
   TARGETDIR "codelets")
 
+set(CODELETS_SRC
+  codelets/codelet_convert.c
+  ${CODELETS_SRC}
+  )
+
 set(RUNTIME_SRCS
   ${RUNTIME_COMMON}
   ${RUNTIME_SRCS_GENERATED}

runtime/starpu/codelets/codelet_convert.c

+/**
+ *
+ * @file starpu/codelet_convert.c
+ *
+ * @copyright 2009-2014 The University of Tennessee and The University of
+ *                      Tennessee Research Foundation. All rights reserved.
+ * @copyright 2012-2023 Bordeaux INP, CNRS (LaBRI UMR 5800), Inria,
+ *                      Univ. Bordeaux. All rights reserved.
+ *
+ ***
+ *
+ * @brief Chameleon zgemm StarPU codelet
+ *
+ * @version 1.3.0
+ * @author Mathieu Faverge
+ * @date 2023-07-06
+ *
+ */
+#include "chameleon_starpu.h"
+#include "runtime_codelets.h"
+#define PRECISION_z
+#include "runtime_codelet_z.h"
+#undef PRECISION_z
+#define PRECISION_d
+#include "runtime_codelet_d.h"
+#undef PRECISION_d
+#define PRECISION_c
+#include "runtime_codelet_c.h"
+#undef PRECISION_c
+#define PRECISION_s
+#include "runtime_codelet_s.h"
+#undef PRECISION_s
+#define PRECISION_zc
+#include "runtime_codelet_zc.h"
+#undef PRECISION_zc
+#define PRECISION_ds
+#include "runtime_codelet_ds.h"
+#undef PRECISION_ds
+
+void
+insert_task_convert( const RUNTIME_option_t *options,
+                     int m, int n,
+                     cham_flttype_t       fromtype,
+                     starpu_data_handle_t fromtile,
+                     cham_flttype_t       totype,
+                     starpu_data_handle_t totile )
+{
+    struct starpu_codelet *codelet = NULL;
+    void (*callback)(void*) = NULL;
+
+    int conversion = ChamConvert( fromtype, totype );
+
+    switch( conversion ) {
+#if defined(CHAMELEON_PREC_ZC)
+    case ChamConvertComplexDoubleToSingle:
+        codelet = &cl_zlag2c;
+        callback = cl_zlag2c_callback;
+        break;
+
+    case ChamConvertComplexSingleToDouble:
+        codelet = &cl_clag2z;
+        callback = cl_clag2z_callback;
+        break;
+#endif
+
+#if defined(CHAMELEON_PREC_DS)
+    case ChamConvertRealDoubleToSingle:
+        codelet = &cl_dlag2s;
+        callback = cl_dlag2s_callback;
+        break;
+
+    case ChamConvertRealSingleToDouble:
+        codelet = &cl_slag2d;
+        callback = cl_slag2d_callback;
+        break;
+#endif
+
+#if defined(CHAMELEON_PREC_D) && defined(CHAMELON_USE_CUDA)
+    case ChamConvertRealDoubleToHalf:
+        codelet = &cl_dlag2h;
+        callback = cl_dlag2h_callback;
+        break;
+
+    case ChamConvertRealHalfToDouble:
+        codelet = &cl_hlag2d;
+        callback = cl_hlag2d_callback;
+        break;
+#endif
+
+#if defined(CHAMELEON_PREC_S) && defined(CHAMELON_USE_CUDA)
+    case ChamConvertRealSingleToHalf:
+        codelet = &cl_slag2h;
+        callback = cl_slag2h_callback;
+        break;
+
+    case ChamConvertRealHalfToSingle:
+        codelet = &cl_hlag2s;
+        callback = cl_hlag2s_callback;
+        break;
+#endif
+
+    case ChamConvertComplexDoubleToDouble:
+        return;
+    case ChamConvertComplexSingleToSingle:
+        return;
+    case ChamConvertComplexHalfToHalf:
+        return;
+    case ChamConvertRealDoubleToDouble:
+        return;
+    case ChamConvertRealSingleToSingle:
+        return;
+    case ChamConvertRealHalfToHalf:
+        return;
+
+    default:
+        assert(0);
+        fprintf( stderr, "INSERT_TASK_convert: Unknown conversion type\n" );
+        return;
+    }
+
+    rt_starpu_insert_task(
+        codelet,
+        STARPU_VALUE,    &m,                 sizeof(int),
+        STARPU_VALUE,    &n,                 sizeof(int),
+        STARPU_R,         fromtile,
+        STARPU_W,         totile,
+        STARPU_PRIORITY,  options->priority,
+        STARPU_CALLBACK,  options->profiling ? callback : NULL,
+        STARPU_EXECUTE_ON_WORKER, options->workerid,
+        0);
+
+    return;
+}

runtime/starpu/include/runtime_codelets.h

@@ -152,4 +152,12 @@
 
 CODELETS_HEADER(map);
 
+void
+insert_task_convert( const RUNTIME_option_t *options,
+                     int m, int n,
+                     cham_flttype_t       fromtype,
+                     starpu_data_handle_t fromtile,
+                     cham_flttype_t       totype,
+                     starpu_data_handle_t totile );
+
 #endif /* _runtime_codelets_h_ */