angelz913 wrote: > Could we add a couple of LIt tests for this? Right now the testing depends on the mlir runner, but most folks probably have it disabled and only run cpu tests in their daily development flow. Added https://github.com/llvm/llvm-project/pull/106082