Mentions légales du service

Skip to content
Snippets Groups Projects
Commit e7d20b3b authored by Mathieu Faverge's avatar Mathieu Faverge
Browse files

Add a mavcro to switch from stream to handle

parent f7e10e33
No related branches found
No related tags found
1 merge request!36Starpu/cublas v2
......@@ -72,6 +72,19 @@ typedef struct starpu_conf starpu_conf_t;
#endif
/*
* cuBlasAPI v2 - StarPU enable the support for cublas handle
*/
#if defined(CHAMELEON_USE_CUDA) && defined(CHAMELEON_USE_CUBLAS_V2)
#define RUNTIME_getStream(_stream_) \
cublasHandle_t _stream_ = starpu_cublas_get_local_handle();
#else
#define RUNTIME_getStream(_stream_) \
cudaStream_t _stream_ = starpu_cuda_get_local_stream(); \
cublasSetKernelStream( stream );
#endif
/*
* Enable codelets names
*/
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment