Reshuffle some code so that functions are defined closer to where they
are used.
Signed-off-by: NThierry Reding <treding@nvidia.com>