Load computations to gpu-memory one at a time

Thanks :slight_smile: