From eae93157f82a463bd9b76d6fa8a33a3ce5fafbc1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jakub=20Klinkovsk=C3=BD?= <klinkjak@fjfi.cvut.cz>
Date: Wed, 14 Oct 2015 18:26:09 +0200
Subject: [PATCH] Made tnl-cuda-arch more robust
(I've actually encountered such error...)
---
tools/src/tnl-cuda-arch.cu | 11 +++++++++--
1 file changed, 9 insertions(+), 2 deletions(-)
diff --git a/tools/src/tnl-cuda-arch.cu b/tools/src/tnl-cuda-arch.cu
index b6bbe8d4dc..f4b8f8d578 100644
--- a/tools/src/tnl-cuda-arch.cu
+++ b/tools/src/tnl-cuda-arch.cu
@@ -1,8 +1,15 @@
#include <stdio.h>
int main() {
- int num_devices;
- cudaGetDeviceCount( &num_devices );
+ int num_devices = 0;
+ cudaError_t error_id = cudaGetDeviceCount( &num_devices );
+
+ if( error_id != cudaSuccess ) {
+ fprintf(stderr, "cudaGetDeviceCount returned error %d (%s)\n",
+ (int) error_id, cudaGetErrorString(error_id));
+ exit(EXIT_FAILURE);
+ }
+
for( int i = 0; i < num_devices; i++ ) {
cudaDeviceProp prop;
cudaGetDeviceProperties( &prop, i );
--
GitLab