starpu_cusparse.h 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2010-2012, 2017 Université de Bordeaux
  4. * Copyright (C) 2010, 2011, 2012, 2013 CNRS
  5. *
  6. * StarPU is free software; you can redistribute it and/or modify
  7. * it under the terms of the GNU Lesser General Public License as published by
  8. * the Free Software Foundation; either version 2.1 of the License, or (at
  9. * your option) any later version.
  10. *
  11. * StarPU is distributed in the hope that it will be useful, but
  12. * WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  14. *
  15. * See the GNU Lesser General Public License in COPYING.LGPL for more details.
  16. */
  17. #ifndef __STARPU_CUSPARSE_H__
  18. #define __STARPU_CUSPARSE_H__
  19. #if defined STARPU_USE_CUDA && !defined STARPU_DONT_INCLUDE_CUDA_HEADERS
  20. #include <cusparse.h>
  21. #endif
  22. #ifdef __cplusplus
  23. extern "C"
  24. {
  25. #endif
  26. void starpu_cusparse_init(void);
  27. void starpu_cusparse_shutdown(void);
  28. #if defined STARPU_USE_CUDA && !defined STARPU_DONT_INCLUDE_CUDA_HEADERS
  29. cusparseHandle_t starpu_cusparse_get_local_handle(void);
  30. #endif
  31. #ifdef __cplusplus
  32. }
  33. #endif
  34. #endif /* __STARPU_CUSPARSE_H__ */