From 176bbfab568f539c07592bac646fdbfcba03cc9c Mon Sep 17 00:00:00 2001
From: Chris Maffeo <cmaffeo2@illinois.edu>
Date: Wed, 6 Jan 2021 11:15:17 -0600
Subject: [PATCH] Fixed -g argument so more than two GPUs can be provided

---
 src/GPUManager.h | 1 +
 src/arbd.cpp     | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/GPUManager.h b/src/GPUManager.h
index e29569b..55b6d63 100644
--- a/src/GPUManager.h
+++ b/src/GPUManager.h
@@ -83,6 +83,7 @@ private:
 	static void init_comms();
 
 public:	
+	static size_t allGpuSize() { return allGpus.size(); }
 	static ncclComm_t* comms;
 	static std::vector<GPU> gpus;
 	
diff --git a/src/arbd.cpp b/src/arbd.cpp
index 5d5b5bc..e326d49 100644
--- a/src/arbd.cpp
+++ b/src/arbd.cpp
@@ -59,7 +59,7 @@ int main(int argc, char* argv[]) {
 	printf("  –––––––––––––––––––––––––––––––––––––––––––––\n");
 	GPUManager::init();
 
-	size_t n_gpus = max(GPUManager::gpus.size(), 1lu);
+	size_t n_gpus = GPUManager::allGpuSize();
 	std::vector<unsigned int> gpuIDs;
 	
 	bool debug = false, safe = false;
-- 
GitLab