StarPU Handbook
starpu_cuda.h
Go to the documentation of this file.
1
/* StarPU --- Runtime system for heterogeneous multicore architectures.
2
*
3
* Copyright (C) 2010-2012,2014 Université de Bordeaux
4
* Copyright (C) 2011 Inria
5
* Copyright (C) 2010-2013,2015,2017,2019 CNRS
6
*
7
* StarPU is free software; you can redistribute it and/or modify
8
* it under the terms of the GNU Lesser General Public License as published by
9
* the Free Software Foundation; either version 2.1 of the License, or (at
10
* your option) any later version.
11
*
12
* StarPU is distributed in the hope that it will be useful, but
13
* WITHOUT ANY WARRANTY; without even the implied warranty of
14
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
15
*
16
* See the GNU Lesser General Public License in COPYING.LGPL for more details.
17
*/
18
19
#ifndef __STARPU_CUDA_H__
20
#define __STARPU_CUDA_H__
21
22
#include <
starpu_config.h
>
23
24
#if defined STARPU_USE_CUDA && !defined STARPU_DONT_INCLUDE_CUDA_HEADERS
25
#include <cuda.h>
26
#include <cuda_runtime.h>
27
#include <cuda_runtime_api.h>
28
29
#ifdef __cplusplus
30
extern
"C"
31
{
32
#endif
33
42
void
starpu_cublas_report_error
(
const
char
*func,
const
char
*file,
int
line,
int
status);
43
47
#define STARPU_CUBLAS_REPORT_ERROR(status) starpu_cublas_report_error(__starpu_func__, __FILE__, __LINE__, status)
48
52
void
starpu_cuda_report_error
(
const
char
*func,
const
char
*file,
int
line, cudaError_t status);
53
57
#define STARPU_CUDA_REPORT_ERROR(status) starpu_cuda_report_error(__starpu_func__, __FILE__, __LINE__, status)
58
70
cudaStream_t
starpu_cuda_get_local_stream
(
void
);
71
76
const
struct
cudaDeviceProp *
starpu_cuda_get_device_properties
(
unsigned
workerid);
77
87
int
starpu_cuda_copy_async_sync
(
void
*src_ptr,
unsigned
src_node,
void
*dst_ptr,
unsigned
dst_node,
size_t
ssize, cudaStream_t stream,
enum
cudaMemcpyKind kind);
88
94
void
starpu_cuda_set_device
(
unsigned
devid);
95
98
#ifdef __cplusplus
99
}
100
#endif
101
102
#endif
/* STARPU_USE_CUDA && !STARPU_DONT_INCLUDE_CUDA_HEADERS */
103
104
#endif
/* __STARPU_CUDA_H__ */
starpu_cuda_set_device
void starpu_cuda_set_device(unsigned devid)
starpu_cuda_report_error
void starpu_cuda_report_error(const char *func, const char *file, int line, cudaError_t status)
starpu_cuda_get_device_properties
const struct cudaDeviceProp * starpu_cuda_get_device_properties(unsigned workerid)
starpu_cublas_report_error
void starpu_cublas_report_error(const char *func, const char *file, int line, int status)
starpu_cuda_copy_async_sync
int starpu_cuda_copy_async_sync(void *src_ptr, unsigned src_node, void *dst_ptr, unsigned dst_node, size_t ssize, cudaStream_t stream, enum cudaMemcpyKind kind)
starpu_config.h
starpu_cuda_get_local_stream
cudaStream_t starpu_cuda_get_local_stream(void)
include
starpu_cuda.h
Generated on Wed May 13 2020 13:17:54 for StarPU Handbook by
1.8.17