瀏覽代碼

Disable using sasum with old cublas, it is buggy with non-blocking stream

Samuel Thibault 8 年之前
父節點
當前提交
594321a4e7
共有 1 個文件被更改,包括 10 次插入2 次删除
  1. 10 2
      tests/microbenchs/matrix_as_vector.c

+ 10 - 2
tests/microbenchs/matrix_as_vector.c

@@ -228,7 +228,6 @@ int main(int argc, char **argv)
 	ret = starpu_init(NULL);
 	if (ret == -ENODEV) return STARPU_TEST_SKIPPED;
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
-	starpu_cublas_init();
 
 	devices = starpu_cpu_worker_get_count();
 	if (devices)
@@ -236,12 +235,21 @@ int main(int argc, char **argv)
 		ret = check_size_on_device(STARPU_CPU, "STARPU_CPU");
 		if (ret) goto error;
 	}
+
+#ifdef STARPU_USE_CUDA
+	cublasHandle_t handle;
+	cublasCreate(&handle);
+	cublasGetVersion(handle, &cublas_version);
+	cublasDestroy(handle);
+
 	devices = starpu_cuda_worker_get_count();
-	if (devices)
+	if (devices && cublas_version >= 7050)
 	{
+		starpu_cublas_init();
 		ret = check_size_on_device(STARPU_CUDA, "STARPU_CUDA");
 		if (ret) goto error;
 	}
+#endif
 #if 0
 	devices = starpu_opencl_worker_get_count();
 	if (devices)