diff --git a/llvm/test/tools/llvm-profgen/inline-cs-pseudoprobe.test b/llvm/test/tools/llvm-profgen/inline-cs-pseudoprobe.test --- a/llvm/test/tools/llvm-profgen/inline-cs-pseudoprobe.test +++ b/llvm/test/tools/llvm-profgen/inline-cs-pseudoprobe.test @@ -9,10 +9,10 @@ ; CHECK-NEXT: 6: 15 ; CHECK-NEXT: 8: 14 bar:14 ; CHECK-NEXT: !CFGChecksum: 138950591924 -; CHECK-NEXT:[main:2 @ foo:8 @ bar]:56:14 +; CHECK-NEXT:[main:2 @ foo:8 @ bar]:28:14 ; CHECK-NEXT: 1: 14 -; CHECK-NEXT: 2: 14 -; CHECK-NEXT: 3: 14 +; CHECK-NEXT: 2: 18446744073709551615 +; CHECK-NEXT: 3: 18446744073709551615 ; CHECK-NEXT: 4: 14 ; CHECK-NEXT: !CFGChecksum: 72617220756 diff --git a/llvm/test/tools/llvm-profgen/noinline-cs-pseudoprobe.test b/llvm/test/tools/llvm-profgen/noinline-cs-pseudoprobe.test --- a/llvm/test/tools/llvm-profgen/noinline-cs-pseudoprobe.test +++ b/llvm/test/tools/llvm-profgen/noinline-cs-pseudoprobe.test @@ -8,10 +8,10 @@ ; CHECK-NEXT: 6: 15 ; CHECK-NEXT: 8: 15 bar:15 ; CHECK-NEXT: !CFGChecksum: 138950591924 -; CHECK-NEXT:[main:2 @ foo:8 @ bar]:60:15 +; CHECK-NEXT:[main:2 @ foo:8 @ bar]:30:15 ; CHECK-NEXT: 1: 15 -; CHECK-NEXT: 2: 15 -; CHECK-NEXT: 3: 15 +; CHECK-NEXT: 2: 18446744073709551615 +; CHECK-NEXT: 3: 18446744073709551615 ; CHECK-NEXT: 4: 15 ; CHECK-NEXT: !CFGChecksum: 72617220756 diff --git a/llvm/tools/llvm-profgen/ProfileGenerator.cpp b/llvm/tools/llvm-profgen/ProfileGenerator.cpp --- a/llvm/tools/llvm-profgen/ProfileGenerator.cpp +++ b/llvm/tools/llvm-profgen/ProfileGenerator.cpp @@ -492,6 +492,22 @@ FunctionSamples &FunctionProfile = getFunctionProfileForLeafProbe(ContextStrStack, Probe, Binary); + // Use InvalidProbeCount(UINT64_MAX) to mark sample count for a dangling + // probe. Dangling probes are the probes associated to an empty block. With + // this place holder, sample count on dangling probe will not be trusted by + // the compiler and it will rely on the counts inference algorithm to get + // the probe a reasonable count. + if (Probe->isDangling()) { + ErrorOr R = FunctionProfile.findSamplesAt(Probe->Index, 0); + if (!R) { + FunctionProfile.addBodySamples(Probe->Index, 0, + FunctionSamples::InvalidProbeCount); + } else { + assert(R.get() == FunctionSamples::InvalidProbeCount && + "Dangling probe count should be UINT64_MAX."); + } + continue; + } FunctionProfile.addBodySamples(Probe->Index, 0, Count); FunctionProfile.addTotalSamples(Count); if (Probe->isEntry()) {