[P] DiscreteFunction does not keep track of space

And thus segfaults if its a temporary. We need to add a corresponding keep alive.