Breaks linalg python tests. Would need to also update python/mlir/dialects/linalg/opdsl. This reverts commit b809d739.