diff --git a/runtime/starpu/include/morse_starpu.h b/runtime/starpu/include/morse_starpu.h
index c1980d3b615f7ef49d51d2797195c7851b2471a7..a98c893e520c98e4c3802d458f2baa5fc6528bcf 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
  */