浏览代码

fix build without CUDA

Samuel Thibault 8 年之前
父节点
当前提交
283d125846
共有 1 个文件被更改,包括 2 次插入0 次删除
  1. 2 0
      src/drivers/cuda/starpu_cublas.c

+ 2 - 0
src/drivers/cuda/starpu_cublas.c

@@ -80,8 +80,10 @@ void starpu_cublas_shutdown(void)
 
 void starpu_cublas_set_stream(void)
 {
+#ifdef STARPU_USE_CUDA
 	if (!_starpu_get_machine_config()->topology.cuda_th_per_dev ||
 		(!_starpu_get_machine_config()->topology.cuda_th_per_stream &&
 		 _starpu_get_machine_config()->topology.nworkerpercuda > 1))
 		cublasSetKernelStream(starpu_cuda_get_local_stream());
+#endif
 }