From e7d20b3ba3bc5625526a0922c4a97803f89e6599 Mon Sep 17 00:00:00 2001
From: Mathieu Faverge <mathieu.faverge@inria.fr>
Date: Thu, 6 Apr 2017 19:28:59 +0200
Subject: [PATCH] Add a mavcro to switch from stream to handle

---
 runtime/starpu/include/morse_starpu.h | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/runtime/starpu/include/morse_starpu.h b/runtime/starpu/include/morse_starpu.h
index c1980d3b6..a98c893e5 100644
--- a/runtime/starpu/include/morse_starpu.h
+++ b/runtime/starpu/include/morse_starpu.h
@@ -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
  */
-- 
GitLab