It would be more of just use MLIR and if you need some extensions for something upstream the changes needed.