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.cu | 11 +++++++++++
 1 file changed, 11 insertions(+)

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

diff --git a/cuda/2d/astra.cu b/cuda/2d/astra.cu
index 81459de..69a4e86 100644
--- a/cuda/2d/astra.cu
+++ b/cuda/2d/astra.cu
@@ -529,5 +529,16 @@ _AstraExport bool setGPUIndex(int iGPUIndex)
         return true;
 }
 
+_AstraExport size_t availableGPUMemory()
+{
+    size_t free, total;
+    cudaError_t err = cudaMemGetInfo(&free, &total);
+    if (err != cudaSuccess)
+        return 0;
+    return free;
+}
+
+
+
 
 }
-- 
cgit v1.2.3