浏览代码

Fix gpu_*register tests when memcpy_peer is explicitly disabled

Samuel Thibault 10 年之前
父节点
当前提交
93de15a034
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 2 2
      tests/datawizard/gpu_ptr_register.c

+ 2 - 2
tests/datawizard/gpu_ptr_register.c

@@ -80,7 +80,7 @@ check_result(unsigned *t, size_t size)
 }
 
 #ifdef STARPU_USE_CUDA
-#if CUDART_VERSION >= 4000
+#ifdef HAVE_CUDA_MEMCPY_PEER
 static int
 test_cuda(void)
 {
@@ -260,7 +260,7 @@ int main(int argc, char **argv)
 #endif
 
 #ifdef STARPU_USE_CUDA
-#if CUDART_VERSION >= 4000 /* We need thread-safety of CUDA */
+#ifdef HAVE_CUDA_MEMCPY_PEER
 	ret = test_cuda();
 	if (ret == 1)
 		goto fail;