'tensor.dim' op unexpected during shape cleanup

[jinchen62]

What happened?

Getting the following error

error: 'tensor.dim' op unexpected during shape cleanup; dynamic dimensions must have been resolved prior to leaving the flow dialect

Full error log: https://gist.github.com/jinchen62/a1de60c0562389f823ddf9d7fe6d24d9
Reproducer: https://gist.github.com/jinchen62/e22cab3ad4562a97c03c8ddfe4f4fd9f

Seems tensor.dim op fails with dynamic shapes in a scf.for loop.

Steps to reproduce your issue

iree-compile --iree-hal-target-backends=llvm-cpu --iree-llvmcpu-target-cpu=host repro.mlir -o test.vmfb

What component(s) does this issue relate to?

No response

Version information

Local build on TOM

Additional context

No response