Add generation step to ddp-in-memory