From fe88a76045c7fd219a6bfff0a946b0ffaf61d471 Mon Sep 17 00:00:00 2001
From: XiaoYun Zhang <xiaoyuz@microsoft.com>
Date: Thu, 18 Jul 2024 14:49:41 -0700
Subject: [PATCH] update

---
 .../Microsoft.ML.GenAI.Phi.Tests.csproj                   | 8 ++++++++
 test/Microsoft.ML.GenAI.Phi.Tests/Phi2Tests.cs            | 7 +++++--
 test/Microsoft.ML.GenAI.Phi.Tests/Phi3Tests.cs            | 7 +++++--
 3 files changed, 18 insertions(+), 4 deletions(-)

diff --git a/test/Microsoft.ML.GenAI.Phi.Tests/Microsoft.ML.GenAI.Phi.Tests.csproj b/test/Microsoft.ML.GenAI.Phi.Tests/Microsoft.ML.GenAI.Phi.Tests.csproj
index 479514a935..1e29582782 100644
--- a/test/Microsoft.ML.GenAI.Phi.Tests/Microsoft.ML.GenAI.Phi.Tests.csproj
+++ b/test/Microsoft.ML.GenAI.Phi.Tests/Microsoft.ML.GenAI.Phi.Tests.csproj
@@ -11,6 +11,7 @@
     <ProjectReference Include="..\..\src\Microsoft.ML.GenAI.Core\Microsoft.ML.GenAI.Core.csproj" />
     <ProjectReference Include="..\..\src\Microsoft.ML.Tokenizers\Microsoft.ML.Tokenizers.csproj" />
     <ProjectReference Include="..\..\src\Microsoft.ML.GenAI.Phi\Microsoft.ML.GenAI.Phi.csproj" />
+    <ProjectReference Include="..\Microsoft.ML.TestFramework\Microsoft.ML.TestFramework.csproj" />
   </ItemGroup>
 
   <ItemGroup>
@@ -38,4 +39,11 @@
     <PackageReference Include="libtorch-cpu-osx-x64" Version="$(LibTorchVersion)" Condition="$([MSBuild]::IsOSPlatform('OSX')) AND '$(TargetArchitecture)' == 'x64'" />
   </ItemGroup>
 
+
+  <ItemGroup>
+    <NativeAssemblyReference Include="MklProxyNative" />
+    <NativeAssemblyReference Include="MklImports" />
+    <NativeAssemblyReference Condition="'$(OS)' == 'Windows_NT'" Include="libiomp5md" />
+  </ItemGroup>
+
 </Project>
diff --git a/test/Microsoft.ML.GenAI.Phi.Tests/Phi2Tests.cs b/test/Microsoft.ML.GenAI.Phi.Tests/Phi2Tests.cs
index b17a74c52e..efd3105ae7 100644
--- a/test/Microsoft.ML.GenAI.Phi.Tests/Phi2Tests.cs
+++ b/test/Microsoft.ML.GenAI.Phi.Tests/Phi2Tests.cs
@@ -8,15 +8,18 @@
 using ApprovalTests.Reporters;
 using FluentAssertions;
 using Microsoft.ML.GenAI.Core.Extension;
+using Microsoft.ML.TestFramework;
 using Microsoft.ML.Tokenizers;
 using TorchSharp;
 using Xunit;
+using Xunit.Abstractions;
 namespace Microsoft.ML.GenAI.Phi.Tests;
 
 [Collection("NoParallelization")]
-public class Phi2Tests
+public class Phi2Tests : BaseTestClass
 {
-    public Phi2Tests()
+    public Phi2Tests(ITestOutputHelper output)
+        : base(output)
     {
         torch.set_default_device("meta");
     }
diff --git a/test/Microsoft.ML.GenAI.Phi.Tests/Phi3Tests.cs b/test/Microsoft.ML.GenAI.Phi.Tests/Phi3Tests.cs
index 7f1b5e5dec..b60a195b8e 100644
--- a/test/Microsoft.ML.GenAI.Phi.Tests/Phi3Tests.cs
+++ b/test/Microsoft.ML.GenAI.Phi.Tests/Phi3Tests.cs
@@ -9,15 +9,18 @@
 using ApprovalTests.Reporters;
 using FluentAssertions;
 using Microsoft.ML.GenAI.Core.Extension;
+using Microsoft.ML.TestFramework;
 using TorchSharp;
 using Xunit;
+using Xunit.Abstractions;
 
 namespace Microsoft.ML.GenAI.Phi.Tests;
 
 [Collection("NoParallelization")]
-public class Phi3Tests
+public class Phi3Tests : BaseTestClass
 {
-    public Phi3Tests()
+    public Phi3Tests(ITestOutputHelper output)
+        : base(output)
     {
         torch.set_default_device("meta");
     }