From 0ad6cb01b9176e99c3e4bb5643ac6a5dd477aad6 Mon Sep 17 00:00:00 2001
From: Willem Jan Palenstijn <Willem.Jan.Palenstijn@cwi.nl>
Date: Thu, 9 Nov 2017 20:57:31 +0100
Subject: Check for CUDA support at run-time in use_cuda()

---
 cuda/2d/astra.h | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'cuda/2d/astra.h')

diff --git a/cuda/2d/astra.h b/cuda/2d/astra.h
index 9355cb4..119d172 100644
--- a/cuda/2d/astra.h
+++ b/cuda/2d/astra.h
@@ -121,5 +121,7 @@ _AstraExport std::string getCudaDeviceString(int device);
 
 _AstraExport bool setGPUIndex(int index);
 
+_AstraExport size_t availableGPUMemory();
+
 }
 #endif
-- 
cgit v1.2.3