riscv: Improve tlb_flush()