fix tag of compile
Previously I'd focused on making sure jitted functions called inside
tagged ones didn't mess with the tagging tree, by propagating that tree
outside argument passing.
That doesn't work when we want to tag values inside jitted parts of the
tagged function, so this change switches things around to propagate the
tagging tree through (e.g. jit) argument passing like ordinary values.
