diff options
author | Robert Maynard <robert.maynard@kitware.com> | 2018-08-15 15:53:51 (GMT) |
---|---|---|
committer | Brad King <brad.king@kitware.com> | 2018-08-23 18:43:31 (GMT) |
commit | 122c79748c30fb80d911591472cba8d6efdb75ae (patch) | |
tree | bbcadf918c0ff404ef3fa22058d305303baf55e7 /Tests/CudaOnly | |
parent | c4ab098097f4fe1bc85ee29c6de390dff435f63f (diff) | |
download | CMake-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.cu | 6 |
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; } |