github
/
flowframes
mirror of https://github.com/n00mkrad/flowframes
1
Fork 0
Code Releases Activity

Cache NCNN GPU check results to avoid crashing rife-ncnn-vulkan on each run

This commit is contained in:
N00MKRAD 2023-12-21 18:23:23 +01:00
parent 89d5cc08a8
commit 979eca6b50
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
Code/Utilities/NcnnUtils.cs
View File

@ -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;
}