Add more ValueTask perf tests (#28937)

Author: stephentoub

…ions/tests/ManualResetValueTaskSource.cs …ks/Sources/ManualResetValueTaskSource.cssrc/System.Threading.Tasks.Extensions/tests/ManualResetValueTaskSource.cs renamed to src/Common/tests/System/Threading/Tasks/Sources/ManualResetValueTaskSource.cs src/System.Threading.Tasks.Extensions/tests/ManualResetValueTaskSource.cs renamed to src/Common/tests/System/Threading/Tasks/Sources/ManualResetValueTaskSource.cs

@@ -3,6 +3,8 @@
 // See the LICENSE file in the project root for more information.
 
 using System.Runtime.CompilerServices;
+using System.Threading.Tasks.Sources;
+using System.Threading.Tasks.Tests;
 using Microsoft.Xunit.Performance;
 using Xunit;
 
@@ -44,6 +46,23 @@ public async Task Await_FromCompletedTask()
             }
         }
 
+        [Benchmark(InnerIterationCount = 10_000_000), MeasureGCAllocations]
+        public async Task Await_FromCompletedValueTaskSource()
+        {
+            ValueTask<int> vt = new ValueTask<int>(ManualResetValueTaskSource.Completed<int>(42), 0);
+            foreach (BenchmarkIteration iteration in Benchmark.Iterations)
+            {
+                long iters = Benchmark.InnerIterationCount;
+                using (iteration.StartMeasurement())
+                {
+                    for (long i = 0; i < iters; i++)
+                    {
+                        await vt;
+                    }
+                }
+            }
+        }
+
         [Benchmark(InnerIterationCount = 10_000_000), MeasureGCAllocations]
         public async Task CreateAndAwait_FromResult()
         {
@@ -110,6 +129,40 @@ public async Task CreateAndAwait_FromCompletedTask_ConfigureAwait()
             }
         }
 
+        [Benchmark(InnerIterationCount = 10_000_000), MeasureGCAllocations]
+        public async Task CreateAndAwait_FromCompletedValueTaskSource()
+        {
+            IValueTaskSource<int> vts = ManualResetValueTaskSource.Completed(42);
+            foreach (BenchmarkIteration iteration in Benchmark.Iterations)
+            {
+                long iters = Benchmark.InnerIterationCount;
+                using (iteration.StartMeasurement())
+                {
+                    for (long i = 0; i < iters; i++)
+                    {
+                        await new ValueTask<int>(vts, 0);
+                    }
+                }
+            }
+        }
+
+        [Benchmark(InnerIterationCount = 10_000_000), MeasureGCAllocations]
+        public async Task CreateAndAwait_FromCompletedValueTaskSource_ConfigureAwait()
+        {
+            IValueTaskSource<int> vts = ManualResetValueTaskSource.Completed(42);
+            foreach (BenchmarkIteration iteration in Benchmark.Iterations)
+            {
+                long iters = Benchmark.InnerIterationCount;
+                using (iteration.StartMeasurement())
+                {
+                    for (long i = 0; i < iters; i++)
+                    {
+                        await new ValueTask<int>(vts, 0).ConfigureAwait(false);
+                    }
+                }
+            }
+        }
+
         [Benchmark(InnerIterationCount = 1_000_000), MeasureGCAllocations]
         public async Task CreateAndAwait_FromYieldingAsyncMethod()
         {
@@ -179,6 +232,23 @@ public void Copy_PassAsArgumentAndReturn_FromTask()
             }
         }
 
+        [Benchmark(InnerIterationCount = 10_000_000), MeasureGCAllocations]
+        public void Copy_PassAsArgumentAndReturn_FromValueTaskSource()
+        {
+            ValueTask<int> vt = new ValueTask<int>(ManualResetValueTaskSource.Completed(42), 0);
+            foreach (BenchmarkIteration iteration in Benchmark.Iterations)
+            {
+                long iters = Benchmark.InnerIterationCount;
+                using (iteration.StartMeasurement())
+                {
+                    for (long i = 0; i < iters; i++)
+                    {
+                        vt = ReturnValueTask(vt);
+                    }
+                }
+            }
+        }
+
         [MethodImpl(MethodImplOptions.NoInlining)]
         private static ValueTask<int> ReturnValueTask(ValueTask<int> vt) => vt;
 

src/System.Threading.Tasks.Extensions/tests/Performance/Perf.ValueTask.cs

@@ -13,6 +13,9 @@
     <Compile Include="$(CommonTestPath)\System\PerfUtils.cs">
       <Link>Common\System\PerfUtils.cs</Link>
     </Compile>
+    <Compile Include="$(CommonTestPath)\System\Threading\Tasks\Sources\ManualResetValueTaskSource.cs">
+      <Link>Common\System\Threading\Tasks\Sources\ManualResetValueTaskSource.cs</Link>
+    </Compile>
   </ItemGroup>
   <ItemGroup>
     <ProjectReference Include="$(CommonPath)\..\perf\PerfRunner\PerfRunner.csproj">

src/System.Threading.Tasks.Extensions/tests/Performance/System.Threading.Tasks.Extensions.Performance.Tests.csproj

@@ -13,8 +13,10 @@
   <ItemGroup>
     <Compile Include="AsyncMethodBuilderAttributeTests.cs" />
     <Compile Include="AsyncValueTaskMethodBuilderTests.cs" />
-    <Compile Include="ManualResetValueTaskSource.cs" />
     <Compile Include="ValueTaskTests.cs" />
+    <Compile Include="$(CommonTestPath)\System\Threading\Tasks\Sources\ManualResetValueTaskSource.cs">
+      <Link>Common\System\Threading\Tasks\Sources\ManualResetValueTaskSource.cs</Link>
+    </Compile>
   </ItemGroup>
   <ItemGroup>
     <Service Include="{82A7F48D-3B50-4B1E-B82E-3ADA8210C358}" />