[U-Boot] [PATCH v3 20/25] riscv: Fix context restore before returning from trap handler