gemm_helper.h 1.1 KB

123456789101112131415161718192021222324252627282930313233343536
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2020 Université de Bordeaux, CNRS (LaBRI UMR 5800), Inria
  4. *
  5. * StarPU is free software; you can redistribute it and/or modify
  6. * it under the terms of the GNU Lesser General Public License as published by
  7. * the Free Software Foundation; either version 2.1 of the License, or (at
  8. * your option) any later version.
  9. *
  10. * StarPU is distributed in the hope that it will be useful, but
  11. * WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  13. *
  14. * See the GNU Lesser General Public License in COPYING.LGPL for more details.
  15. */
  16. #ifndef __MPI_TESTS_GEMM_HELPER__
  17. #define __MPI_TESTS_GEMM_HELPER__
  18. #include <starpu_config.h>
  19. extern unsigned nslices;
  20. extern unsigned matrix_dim;
  21. extern unsigned check;
  22. extern int comm_thread_cpuid;
  23. void gemm_alloc_data();
  24. int gemm_init_data();
  25. int gemm_submit_tasks();
  26. void gemm_release();
  27. void gemm_add_polling_dependencies();
  28. int gemm_submit_tasks_with_tags(int with_tags);
  29. #endif /* __MPI_TESTS_GEMM_HELPER__ */