update

dotnet · Jul 18, 2024 · fe88a76 · fe88a76
1 parent b316f19
commit fe88a76
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 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
@@ -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
@@ -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
@@ -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");
     }