From 9832489dc2830cbdce42deea77857b7979904761 Mon Sep 17 00:00:00 2001
From: David Mayerich <mayerich@uh.edu>
Date: Tue, 10 Jul 2018 13:26:15 -0500
Subject: [PATCH] edited main.cu to make the program more robust if a GPU isn't available

---
 main.cu | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/main.cu b/main.cu
index fcaab35..3d2292b 100644
--- a/main.cu
+++ b/main.cu
@@ -822,8 +822,11 @@ void setdevice(int &device){
 	int count;
 	cudaGetDeviceCount(&count);					// numbers of device that are available
 	if(count < device + 1){
-	std::cout<<"No such device available, please set another device"<<std::endl;
-	exit(1);
+		std::cout<<"The selected CUDA device ("<<device<<") is unavailable. "<<count<<" devices are detected."<<std::endl;
+		if (count == 0)
+			device = -1;
+		else
+			exit(1);
 	}
 }
 #else
--
libgit2 0.21.4