PRU code update

Tweak PRU code.

Signed-off-by: Paul Brook <paul@nowt.org>