While not really answering the question on why the shader misbehaves, I did find a solution for my problem.

It comes down to me misunderstanding how stencil tests work. It turns out you can control stencil buffer writes by discarding the fragment in the fragment shader. This lets me have one subpass that writes to the stencil buffer in appropriate places and another one after it that uses to discard unneeded fragments.