Add GPU lease warm-up and shared limiter
This commit is contained in:
1 parent
0e7bbae132
commit
00d4e9bde2
4 files changed
+139
-48
No files matched your search
@@ -0,0 +1,37 @@
|
||||
using System.Threading;
|
||||
|
||||
namespace PerfectNumbers.Core.Gpu;
|
||||
|
||||
public static class GpuPrimeLeaseLimiter
|
||||
{
|
||||
private static SemaphoreSlim _semaphore = new(int.MaxValue, int.MaxValue);
|
||||
private static int _currentLimit = int.MaxValue;
|
||||
|
||||
public static void Initialize(int limit)
|
||||
{
|
||||
if (limit <= 0)
|
||||
{
|
||||
limit = 1;
|
||||
}
|
||||
|
||||
if (limit == _currentLimit)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
var previous = _semaphore;
|
||||
_semaphore = new SemaphoreSlim(limit, limit);
|
||||
_currentLimit = limit;
|
||||
previous.Dispose();
|
||||
}
|
||||
|
||||
public static void Enter()
|
||||
{
|
||||
_semaphore.Wait();
|
||||
}
|
||||
|
||||
public static void Exit()
|
||||
{
|
||||
_semaphore.Release();
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user