diff options
| author | Pierre Moreau <dev@pmoreau.org> | 2017-02-08 21:50:58 (GMT) |
|---|---|---|
| committer | Brad King <brad.king@kitware.com> | 2017-02-10 18:47:05 (GMT) |
| commit | 166b3393875e4f6752907b9b06c2518dce0432e9 (patch) | |
| tree | 9a1484f0b5c34c34decbc8dc4ae6796a4a43f911 /Tests/Cuda/Complex/main.cpp | |
| parent | cbe4d5957b526f9a9bb6954d0c6944ad7642bb90 (diff) | |
| download | CMake-166b3393875e4f6752907b9b06c2518dce0432e9.zip CMake-166b3393875e4f6752907b9b06c2518dce0432e9.tar.gz CMake-166b3393875e4f6752907b9b06c2518dce0432e9.tar.bz2 | |
Tests/Cuda: Select a CUDA device supporting compute 3.0
Diffstat (limited to 'Tests/Cuda/Complex/main.cpp')
| -rw-r--r-- | Tests/Cuda/Complex/main.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/Tests/Cuda/Complex/main.cpp b/Tests/Cuda/Complex/main.cpp index 92d1fb0..2498235 100644 --- a/Tests/Cuda/Complex/main.cpp +++ b/Tests/Cuda/Complex/main.cpp @@ -9,11 +9,17 @@ #define IMPORT #endif +IMPORT int choose_cuda_device(); IMPORT int call_cuda_seperable_code(int x); IMPORT int mixed_launch_kernel(int x); int main(int argc, char** argv) { + int ret = choose_cuda_device(); + if (ret) { + return 0; + } + int r1 = call_cuda_seperable_code(42); int r2 = mixed_launch_kernel(42); return (r1 == 42 || r2 == 42) ? 1 : 0; |