Create HLO token in the computation only when needed

[jonatanklosko]

We now check if the token will be needed, and only create if so. In places that pass the token through (optional computation, while, ...), we now do so conditionally.