diff --git a/tests/helper/cublasLt_init.c b/tests/helper/cublasLt_init.c new file mode 100644 index 0000000000..6c4971ee76 --- /dev/null +++ b/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 +#include +#include +#include +#include +#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; +}