nir/scheduler: Add an option to specify what stages share memory for I/O