From 4f47b4a30f58086fd16b70de593beb89b4a4d710 Mon Sep 17 00:00:00 2001
From: Mathieu Faverge <mathieu.faverge@inria.fr>
Date: Sat, 27 May 2023 20:52:08 -0400
Subject: [PATCH] starpu/codelets/convert: add a conversion codelet to
automatically change the precision on request
---
runtime/starpu/CMakeLists.txt | 5 +
runtime/starpu/codelets/codelet_convert.c | 133 ++++++++++++++++++++++
runtime/starpu/include/runtime_codelets.h | 8 ++
3 files changed, 146 insertions(+)
create mode 100644 runtime/starpu/codelets/codelet_convert.c
diff --git a/runtime/starpu/CMakeLists.txt b/runtime/starpu/CMakeLists.txt
index 9711e7934..3e8189905 100644
--- a/runtime/starpu/CMakeLists.txt
+++ b/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}
diff --git a/runtime/starpu/codelets/codelet_convert.c b/runtime/starpu/codelets/codelet_convert.c
new file mode 100644
index 000000000..0b07c069a
--- /dev/null
+++ b/runtime/starpu/codelets/codelet_convert.c
@@ -0,0 +1,133 @@
+/**
+ *
+ * @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;
+}
diff --git a/runtime/starpu/include/runtime_codelets.h b/runtime/starpu/include/runtime_codelets.h
index a8540b0cd..f89b77293 100644
--- a/runtime/starpu/include/runtime_codelets.h
+++ b/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_ */
--
GitLab