diff --git a/arch/loongarch/mm/tlb.c b/arch/loongarch/mm/tlb.c
index 9818ce11546bcb502415dbe7e000d1798ae4aac4..e18e2aadab657ba03fa3a34cce24a6222cd27ea1 100644
--- a/arch/loongarch/mm/tlb.c
+++ b/arch/loongarch/mm/tlb.c
@@ -258,6 +258,7 @@ extern long exception_handlers[VECSIZE * 128 / sizeof(long)];
 void setup_tlb_handler(int cpu)
 {
 	setup_ptwalker();
+	local_flush_tlb_all();
 	output_pgtable_bits_defines();
 
 	/* The tlb handlers are generated only once */
@@ -302,5 +303,4 @@ void tlb_init(int cpu)
 	write_csr_stlbpgsize(PS_DEFAULT_SIZE);
 	write_csr_tlbrefill_pagesize(PS_DEFAULT_SIZE);
 	setup_tlb_handler(cpu);
-	local_flush_tlb_all();
 }