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