From 979eca6b501ada586d6287bfc97e7b01034c654c Mon Sep 17 00:00:00 2001
From: N00MKRAD <61149547+n00mkrad@users.noreply.github.com>
Date: Thu, 21 Dec 2023 18:23:23 +0100
Subject: [PATCH] Cache NCNN GPU check results to avoid crashing
 rife-ncnn-vulkan on each run

---
 Code/Utilities/NcnnUtils.cs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/Code/Utilities/NcnnUtils.cs b/Code/Utilities/NcnnUtils.cs
index 7b88a7a..c016eed 100644
--- a/Code/Utilities/NcnnUtils.cs
+++ b/Code/Utilities/NcnnUtils.cs
@@ -8,16 +8,20 @@ using System.Diagnostics;
 using System.Drawing;
 using System.IO;
 using System.Linq;
-using System.Text;
 using System.Threading.Tasks;
 
 namespace Flowframes.Utilities
 {
     class NcnnUtils
     {
+        public static Dictionary<int, int> GpuIdsQueueCounts = null;
+
         /// <summary> Get amount of GPU Compute Queues (VK) for each GPU </summary>
         public static async Task<Dictionary<int, int>> GetNcnnGpuComputeQueueCounts ()
         {
+            if(GpuIdsQueueCounts != null)
+                return GpuIdsQueueCounts;
+
             Dictionary<int, int> queueCounts = new Dictionary<int, int>(); // int gpuId, int queueCount
 
             Process rifeNcnn = OsUtils.NewProcess(true);
@@ -34,6 +38,7 @@ namespace Flowframes.Utilities
                 queueCounts[gpuId] = queueCount;
             }
 
+            GpuIdsQueueCounts = queueCounts;
             return queueCounts;
         }