The generative graph doesn’t mix info, so the derived logp is correct. ndim_supp doesn’t have an effect there are you found out.
Instead what you can do is overwrite the logp and sum the last entries. We do that for the logp of RandomWalk if you want a code example