Strange error with rewrite each, possible due to inconsistent state

I get a strange error when using `rewrite each`. I suspect that it's because the state is already inconsistent but Pulse hasn't realized and the rewrite triggers a consistency check which fails (just speculating though). I've pushed the code with the issue to the `inconsistent-state` branch. The problematic code is here: https://github.com/mtzguido/gpuver/blob/inconsistent-state/src/examples/matmul-opt/GPU.MatMulOpt.Kernel.fsti#L150-L151 removing it or also strangely moving those lines to the start of the function avoids the error.