TrivialLocalMem optimization is meant for entry function only. Disable it for subroutines.

Change-Id: I77854c0c055cf7936d4df43b0b5bb0bfd0c87485

Author: jgu222

IGC/Compiler/CustomSafeOptPass.cpp

@@ -1632,6 +1632,14 @@ bool TrivialLocalMemoryOpsElimination::runOnFunction(Function& F)
 {
     bool change = false;
 
+    IGCMD::MetaDataUtils* pMdUtil = getAnalysis<MetaDataUtilsWrapper>().getMetaDataUtils();
+    if (!isEntryFunc(pMdUtil, &F))
+    {
+        // Skip if it is non-entry function.  For example, a subroutine
+        //   foo ( local int* p) { ...... store v, p; ......}
+        // in which no localMemoptimization will be performed.
+        return change;
+    }
 
     visit(F);
     if (!abortPass && (m_LocalLoadsToRemove.empty() ^ m_LocalStoresToRemove.empty()))

IGC/Compiler/CustomSafeOptPass.hpp

@@ -26,6 +26,7 @@ SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 
 #pragma once
 #include "Compiler/CodeGenContextWrapper.hpp"
+#include "Compiler/MetaDataUtilsWrapper.h"
 #include "common/MDFrameWork.h"
 #include "common/LLVMWarningsPush.hpp"
 #include "llvm/Config/llvm-config.h"
@@ -124,6 +125,7 @@ namespace IGC
         virtual void getAnalysisUsage(llvm::AnalysisUsage& AU) const override
         {
             AU.addRequired<CodeGenContextWrapper>();
+            AU.addRequired<MetaDataUtilsWrapper>();
             AU.setPreservesCFG();
         }