From ffb01f6e8a4af923aa5e04fbc91e175562b663e3 Mon Sep 17 00:00:00 2001
From: Ching-Hsin Lee <chinglee@amazon.com>
Date: Tue, 11 Apr 2023 17:53:05 +0800
Subject: [PATCH] Not yield for running task in prvYieldForTask

* Raise priority of a running task should not alter other cores
---
 tasks.c | 33 +++++++++++++++++++++------------
 1 file changed, 21 insertions(+), 12 deletions(-)

diff --git a/tasks.c b/tasks.c
index 5ee298848b9..840c331169b 100644
--- a/tasks.c
+++ b/tasks.c
@@ -795,6 +795,10 @@ static void prvAddNewTaskToReadyList( TCB_t * pxNewTCB ) PRIVILEGED_FUNCTION;
             /* No task should yield for this one if it is a lower priority
              * than priority level of currently ready tasks. */
             if( pxTCB->uxPriority >= uxTopReadyPriority )
+        #else
+
+            /* Yield is not required for a task which is already running. */
+            if( taskTASK_IS_RUNNING( pxTCB ) == pdFALSE )
         #endif
         {
             xLowestPriorityToPreempt = ( BaseType_t ) pxTCB->uxPriority;
@@ -815,24 +819,29 @@ static void prvAddNewTaskToReadyList( TCB_t * pxNewTCB ) PRIVILEGED_FUNCTION;
 
                 if( ( taskTASK_IS_RUNNING( pxCurrentTCBs[ xCoreID ] ) != pdFALSE ) && ( xYieldPendings[ xCoreID ] == pdFALSE ) )
                 {
-                    if( xCurrentCoreTaskPriority <= xLowestPriorityToPreempt )
+                    #if ( configRUN_MULTIPLE_PRIORITIES == 0 )
+                        if( taskTASK_IS_RUNNING( pxTCB ) == pdFALSE )
+                    #endif
                     {
-                        #if ( configUSE_CORE_AFFINITY == 1 )
-                            if( ( pxTCB->uxCoreAffinityMask & ( 1 << xCoreID ) ) != 0 )
-                        #endif
+                        if( xCurrentCoreTaskPriority <= xLowestPriorityToPreempt )
                         {
-                            #if ( configUSE_TASK_PREEMPTION_DISABLE == 1 )
-                                if( pxCurrentTCBs[ xCoreID ]->xPreemptionDisable == pdFALSE )
+                            #if ( configUSE_CORE_AFFINITY == 1 )
+                                if( ( pxTCB->uxCoreAffinityMask & ( 1 << xCoreID ) ) != 0 )
                             #endif
                             {
-                                xLowestPriorityToPreempt = xCurrentCoreTaskPriority;
-                                xLowestPriorityCore = xCoreID;
+                                #if ( configUSE_TASK_PREEMPTION_DISABLE == 1 )
+                                    if( pxCurrentTCBs[ xCoreID ]->xPreemptionDisable == pdFALSE )
+                                #endif
+                                {
+                                    xLowestPriorityToPreempt = xCurrentCoreTaskPriority;
+                                    xLowestPriorityCore = xCoreID;
+                                }
                             }
                         }
-                    }
-                    else
-                    {
-                        mtCOVERAGE_TEST_MARKER();
+                        else
+                        {
+                            mtCOVERAGE_TEST_MARKER();
+                        }
                     }
 
                     #if ( configRUN_MULTIPLE_PRIORITIES == 0 )