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