Mirrors/flowframes
1
0
Fork 0
mirror of https://github.com/n00mkrad/flowframes.git synced 2025-12-22 03:09:28 +01:00
Code Issues Packages Projects Releases Wiki Activity

Increase default NCNN threads, allow more than 3 with NCNN VS

Browse Source
This commit is contained in:
n00mkrad
2023-02-17 22:10:33 +01:00
parent fbee4d8c06
commit c20c1b55ee
3 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
Code/Utilities/NcnnUtils.cs
View File

@@ -39,7 +39,7 @@ namespace Flowframes.Utilities
public static async Task<int> GetRifeNcnnGpuThreads(Size res, int gpuId, AI ai)
{
int threads = 8;
int threads = Config.GetInt(Config.Key.ncnnThreads);
//if (res.Width * res.Height > 2560 * 1440) threads = 4;
// if (res.Width * res.Height > 3840 * 2160) threads = 1;
@@ -74,14 +74,14 @@ namespace Flowframes.Utilities
return tilesizeStr;
}
public static string GetNcnnThreads(bool forceSingleThread = false)
public static async Task<string> GetNcnnThreads(AI ai)
{
int gpusAmount = Config.Get(Config.Key.ncnnGpus).Split(',').Length;
int procThreads = Config.GetInt(Config.Key.ncnnThreads);
string progThreadsStr = $"{procThreads}";
int threads = await GetRifeNcnnGpuThreads(new Size(), Config.Get(Config.Key.ncnnGpus).Split(',')[0].GetInt(), ai);
string progThreadsStr = $"{threads}";
for (int i = 1; i < gpusAmount; i++)
progThreadsStr += $",{procThreads}";
progThreadsStr += $",{threads}";
return $"{(forceSingleThread ? 1 : (Interpolate.currentlyUsingAutoEnc ? 2 : 4))}:{progThreadsStr}:4"; // Read threads: 1 for singlethreaded, 2 for autoenc, 4 if order is irrelevant
}