[U-Boot] [PATCH 14/19] riscv: Fix context restore before returning from trap handler