Add test for cublasLt

tests/helper/cublasLt_init.c

@@ -0,0 +1,71 @@
+/* StarPU --- Runtime system for heterogeneous multicore architectures.
+*
+ * Copyright (C) 2009-2024  University of Bordeaux, CNRS (LaBRI UMR 5800), Inria
+ *
+ * StarPU is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation; either version 2.1 of the License, or (at
+ * your option) any later version.
+ *
+ * StarPU is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
+ *
+ * See the GNU Lesser General Public License in COPYING.LGPL for more details.
+ */
+
+#include <stdio.h>
+#include <unistd.h>
+#include <errno.h>
+#include <starpu.h>
+#include <stdlib.h>
+#include "../helper.h"
+
+/*
+ * Test initializing cublasLt, and how much time that takes
+ */
+
+static double start;
+static double end;
+
+//static float *data = NULL;
+
+int main(int argc, char **argv)
+{
+    int ret;
+
+    ret = starpu_initialize(NULL, &argc, &argv);
+    if (ret == -ENODEV) return STARPU_TEST_SKIPPED;
+    STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
+
+    unsigned ngpus = starpu_cuda_worker_get_count();
+
+    double init_timing;
+    double shutdown_timing;
+
+    start = starpu_timing_now();
+    starpu_cublasLt_init();
+    end = starpu_timing_now();
+    init_timing = end - start;
+
+    start = starpu_timing_now();
+    starpu_cublasLt_shutdown();
+    end = starpu_timing_now();
+    shutdown_timing = end - start;
+
+    FPRINTF(stderr, "Total:\n");
+    FPRINTF(stderr, "\tinit: %2.2f ms\n", init_timing/(1000));
+    FPRINTF(stderr, "\tshutdown: %2.2f ms\n", shutdown_timing/(1000));
+
+    if (ngpus != 0)
+    {
+        FPRINTF(stderr, "per-GPU (#gpu = %u):\n", ngpus);
+
+        FPRINTF(stderr, "\tinit: %2.2f ms\n", init_timing/(1000*ngpus));
+        FPRINTF(stderr, "\tshutdown: %2.2f ms\n", shutdown_timing/(1000*ngpus));
+    }
+
+    starpu_shutdown();
+
+    return EXIT_SUCCESS;
+}