From 48ef3bb08058141c8e5c1a2087e3d64184cfc7c2 Mon Sep 17 00:00:00 2001
From: Martin Evans <martindevans@gmail.com>
Date: Tue, 2 Jan 2024 19:14:21 +0000
Subject: [PATCH] Added runtime checks that UseMemoryLock and UseMemorymap are
 actually supported.

---
 LLama/Extensions/IModelParamsExtensions.cs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/LLama/Extensions/IModelParamsExtensions.cs b/LLama/Extensions/IModelParamsExtensions.cs
index 36558e72..f3ef31f3 100644
--- a/LLama/Extensions/IModelParamsExtensions.cs
+++ b/LLama/Extensions/IModelParamsExtensions.cs
@@ -21,10 +21,14 @@ public static class IModelParamsExtensions
     /// <exception cref="ArgumentException"></exception>
     public static IDisposable ToLlamaModelParams(this IModelParams @params, out LLamaModelParams result)
     {
+        if (@params.UseMemoryLock && !NativeApi.llama_mlock_supported())
+            throw new NotSupportedException("'UseMemoryLock' is not supported (llama_mlock_supported() == false)");
+        if (@params.UseMemorymap && !NativeApi.llama_mmap_supported())
+            throw new NotSupportedException("'UseMemorymap' is not supported (llama_mmap_supported() == false)");
+
         var disposer = new GroupDisposable();
 
         result = NativeApi.llama_model_default_params();
-
         result.main_gpu = @params.MainGpu;
         result.n_gpu_layers = @params.GpuLayerCount;
         result.use_mlock = @params.UseMemoryLock;