summaryrefslogtreecommitdiffstats
path: root/Tests/CudaOnly
diff options
context:
space:
mode:
authorRobert Maynard <robert.maynard@kitware.com>2018-08-15 15:53:51 (GMT)
committerBrad King <brad.king@kitware.com>2018-08-23 18:43:31 (GMT)
commit122c79748c30fb80d911591472cba8d6efdb75ae (patch)
treebbcadf918c0ff404ef3fa22058d305303baf55e7 /Tests/CudaOnly
parentc4ab098097f4fe1bc85ee29c6de390dff435f63f (diff)
downloadCMake-122c79748c30fb80d911591472cba8d6efdb75ae.zip
CMake-122c79748c30fb80d911591472cba8d6efdb75ae.tar.gz
CMake-122c79748c30fb80d911591472cba8d6efdb75ae.tar.bz2
CUDA: Avoid using deprecated cublas_device to identify device lib dirs
Use `curand_static` to identify directories containing device libraries because `cublas_device` is deprecated and will be removed in the future. Issue: #18290
Diffstat (limited to 'Tests/CudaOnly')
-rw-r--r--Tests/CudaOnly/LinkSystemDeviceLibraries/main.cu6
1 files changed, 6 insertions, 0 deletions
diff --git a/Tests/CudaOnly/LinkSystemDeviceLibraries/main.cu b/Tests/CudaOnly/LinkSystemDeviceLibraries/main.cu
index 7eecec1..2c7c388 100644
--- a/Tests/CudaOnly/LinkSystemDeviceLibraries/main.cu
+++ b/Tests/CudaOnly/LinkSystemDeviceLibraries/main.cu
@@ -3,6 +3,9 @@
#include <cuda_runtime.h>
#include <iostream>
+// this test only makes sense for versions of CUDA that ships
+// static libraries that have separable compilation device symbols
+#if __CUDACC_VER_MAJOR__ <= 9
__global__ void deviceCublasSgemm(int n, float alpha, float beta,
const float* d_A, const float* d_B,
float* d_C)
@@ -22,6 +25,7 @@ __global__ void deviceCublasSgemm(int n, float alpha, float beta,
cublasDestroy(cnpHandle);
}
+#endif
int choose_cuda_device()
{
@@ -63,6 +67,7 @@ int main(int argc, char** argv)
return 0;
}
+#if __CUDACC_VER_MAJOR__ <= 9
// initial values that will make sure that the cublasSgemm won't actually
// do any work
int n = 0;
@@ -72,6 +77,7 @@ int main(int argc, char** argv)
float* d_B = nullptr;
float* d_C = nullptr;
deviceCublasSgemm<<<1, 1>>>(n, alpha, beta, d_A, d_B, d_C);
+#endif
return 0;
}