Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106