From fbe4fd4ea10faad7ce1303a16425ab9512cfebea Mon Sep 17 00:00:00 2001
From: Rinne <AsakusaRinne@gmail.com>
Date: Wed, 8 May 2024 21:17:04 +0800
Subject: [PATCH] debug

---
 LLama.Benchmark/LLama.Benchmark.csproj            | 1 +
 LLama.Benchmark/LLamaExecutorBenchmark/Prefill.cs | 5 +++--
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/LLama.Benchmark/LLama.Benchmark.csproj b/LLama.Benchmark/LLama.Benchmark.csproj
index 6e400e4c..17828415 100644
--- a/LLama.Benchmark/LLama.Benchmark.csproj
+++ b/LLama.Benchmark/LLama.Benchmark.csproj
@@ -11,6 +11,7 @@
 
   <ItemGroup>
     <PackageReference Include="BenchmarkDotNet" Version="0.13.12" />
+    <PackageReference Include="BenchmarkDotNet.Diagnostics.Windows" Version="0.13.12" />
   </ItemGroup>
 
     <ItemGroup>
diff --git a/LLama.Benchmark/LLamaExecutorBenchmark/Prefill.cs b/LLama.Benchmark/LLamaExecutorBenchmark/Prefill.cs
index 6b9dcf5e..4d9df0aa 100644
--- a/LLama.Benchmark/LLamaExecutorBenchmark/Prefill.cs
+++ b/LLama.Benchmark/LLamaExecutorBenchmark/Prefill.cs
@@ -11,13 +11,14 @@ using BenchmarkDotNet.Engines;
 using BenchmarkDotNet.Jobs;
 using LLama.Abstractions;
 using LLama.Common;
-using BenchmarkDotNet.Diagnostics.Windows.Configs;
 
 namespace LLama.Benchmark.LLamaExecutorBenchmark
 {
+#if WINDOWS
+    [BenchmarkDotNet.Diagnostics.Windows.Configs.NativeMemoryProfiler]
+#endif
     [BenchmarkCategory("Executor", "LLama")]
     [SimpleJob(RunStrategy.ColdStart, runtimeMoniker: RuntimeMoniker.Net80)]
-    [NativeMemoryProfiler]
     [MemoryDiagnoser]
     [MinIterationCount(1)]
     [MaxIterationCount(16)]