Balanced Source Generator mode does not work on saving

[Dreamescaper]

Based on the description, source generators should be invoked on document Save. However, this doesn't happen.

Version Used:
Visual Studio 2022 v17.12 Preview 4

Steps to Reproduce:
Paste the snippet to Visual Studio:

using System.Text.RegularExpressions;

var regex = MyTestRegex();
regex.Match("1234");

partial class Program
{
    [GeneratedRegex(@"\w+")]
    private static partial Regex MyTestRegex();
}

Click Save / Press Ctrl+S.

Expected Behavior:
Method implemenation is generated, and the error is not displayed.

Actual Behavior:
Method implementation is not generated until build.

CS8795 Partial method 'Program.MyTestRegex()' must have an implementation part because it has accessibility modifiers.

Error disappears after build.

[Dreamescaper]

[Dreamescaper]

@CyrusNajmabadi
I've re-checked this with 17.12.2.
Apparently, it does generate the code on Save, but VS doesn't pick up changes, and shows an error. So even if I see the generated file, and even IntelliSense shows a proper description, VS still considers it as non-implemented until build.

It does not reproduce with Automatic mode, so it still seems to be Balanced mode related.

Let me know if you need any help reproducing it.

[CyrusNajmabadi]

I think this may be a pull diagnostics issue @dibarbet . We are rerunning generators. And we can successfully even navigate to the generated code. But the diagnostics shows the results as if the generated code is not being analyzed. Perhaps a caching issue where we're not properly invalidating and rerunning pull diags in the presence of changed SG results?