Generating High-Performance Schedules in MLIR