|
@@ -1,6 +1,6 @@
|
|
|
|
|
|
*
|
|
|
- * Copyright (C) 2009, 2010-2011, 2015 Université de Bordeaux
|
|
|
+ * Copyright (C) 2009, 2010-2011, 2015, 2017 Université de Bordeaux
|
|
|
* Copyright (C) 2010, 2011 CNRS
|
|
|
*
|
|
|
* StarPU is free software; you can redistribute it and/or modify
|
|
@@ -48,12 +48,13 @@ static inline void common_block_spmv(void *descr[], int s, STARPU_ATTRIBUTE_UNUS
|
|
|
cblas_sgemv(CblasRowMajor, CblasNoTrans, dx, dy, 1.0f, block, ld, in, 1, 1.0f, out, 1);
|
|
|
break;
|
|
|
#ifdef STARPU_USE_CUDA
|
|
|
- case 1:
|
|
|
+ case 1: {
|
|
|
cublasStatus_t status = cublasSgemv (starpu_cublas_get_local_handle(),
|
|
|
CUBLAS_OP_T, dx, dy, &p1, block, ld, in, 1, &p1, out, 1);
|
|
|
if (status != CUBLAS_STATUS_SUCCESS)
|
|
|
STARPU_CUBLAS_REPORT_ERROR(status);
|
|
|
break;
|
|
|
+ }
|
|
|
#endif
|
|
|
default:
|
|
|
STARPU_ABORT();
|