2 files changed, 3 insertions, 18 deletions

diff --git a/arch/nios2/kernel/cpuinfo.c b/arch/nios2/kernel/cpuinfo.c
index 51d5bb90d3e5..a223691dff4f 100644
--- a/arch/nios2/kernel/cpuinfo.c
+++ b/arch/nios2/kernel/cpuinfo.c
@@ -72,6 +72,7 @@ void __init setup_cpuinfo(void)
         cpuinfo.has_div = fcpu_has(cpu, "altr,has-div");
         cpuinfo.has_mul = fcpu_has(cpu, "altr,has-mul");
         cpuinfo.has_mulx = fcpu_has(cpu, "altr,has-mulx");
+        cpuinfo.mmu = fcpu_has(cpu, "altr,has-mmu");
 
         if (IS_ENABLED(CONFIG_NIOS2_HW_DIV_SUPPORT) && !cpuinfo.has_div)
                 err_cpu("DIV");
diff --git a/arch/nios2/kernel/entry.S b/arch/nios2/kernel/entry.S
index 83bca17d1008..0bdfd13ff98b 100644
--- a/arch/nios2/kernel/entry.S
+++ b/arch/nios2/kernel/entry.S
@@ -365,30 +365,14 @@ ENTRY(ret_from_interrupt)
         GET_THREAD_INFO r1
         ldw     r4, TI_PREEMPT_COUNT(r1)
         bne     r4, r0, restore_all
-
-need_resched:
         ldw     r4, TI_FLAGS(r1)                /* ? Need resched set */
         BTBZ    r10, r4, TIF_NEED_RESCHED, restore_all
         ldw     r4, PT_ESTATUS(sp)      /* ? Interrupts off */
         andi    r10, r4, ESTATUS_EPIE
         beq     r10, r0, restore_all
-        movia   r4, PREEMPT_ACTIVE
-        stw     r4, TI_PREEMPT_COUNT(r1)
-        rdctl   r10, status             /* enable intrs again */
-        ori     r10, r10 ,STATUS_PIE
-        wrctl   status, r10
-        PUSH    r1
-        call    schedule
-        POP     r1
-        mov     r4, r0
-        stw     r4, TI_PREEMPT_COUNT(r1)
-        rdctl   r10, status             /* disable intrs */
-        andi    r10, r10, %lo(~STATUS_PIE)
-        wrctl   status, r10
-        br      need_resched
-#else
-        br      restore_all
+        call    preempt_schedule_irq
 #endif
+        br      restore_all
 
 /***********************************************************************
  * A few syscall wrappers