[mlir][linalg] Cast back to the original type after making linalg.generic outputs more static

This codepath was entirely untested.

Differential Revision: https://reviews.llvm.org/D120473