[Mosaic GPU] Add a `fuel` argument to `infer_layout`.
This allows us to restrict the maximum number of search steps taken during
compilation. We may fail because the value is not high enough, but it'll
avoid the impression that we're "deadlocking" when the search is not finding
a solution.
This ought to be a better user experience.
PiperOrigin-RevId: 827832664
