Preserve cache policy metadata

Changed some load optimization passes so that they preserve the lsc.cache.ctrl metadata that is used to override cache policy in the HW.
intel · Aug 5, 2024 · bbaecb1 · bbaecb1
1 parent f21497a
commit bbaecb1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/IGC/Compiler/CISACodeGen/LdShrink.cpp b/IGC/Compiler/CISACodeGen/LdShrink.cpp
@@ -146,6 +146,10 @@ bool LdShrink::runOnFunction(Function& F) {
 
             LoadInst* NewLoad = Builder.CreateAlignedLoad(ScalarTy, ScalarPtr, IGCLLVM::getAlign(alignment));
             NewLoad->setDebugLoc(LI->getDebugLoc());
+            if (MDNode* mdNode = LI->getMetadata("lsc.cache.ctrl"))
+            {
+                NewLoad->setMetadata("lsc.cache.ctrl", mdNode);
+            }
 
             ExtractElementInst* EEI = cast<ExtractElementInst>(*LI->user_begin());
             EEI->replaceAllUsesWith(NewLoad);

diff --git a/IGC/Compiler/CISACodeGen/VectorPreProcess.cpp b/IGC/Compiler/CISACodeGen/VectorPreProcess.cpp
@@ -150,7 +150,12 @@ namespace
             {
                 Type* newPtrType = PointerType::get(returnType, ptr->getType()->getPointerAddressSpace());
                 ptr = builder.CreateBitCast(ptr, newPtrType);
-                return builder.CreateAlignedLoad(returnType, ptr, IGCLLVM::getAlign(alignment), isVolatile);
+                LoadInst* newLI = builder.CreateAlignedLoad(returnType, ptr, IGCLLVM::getAlign(alignment), isVolatile);
+                if (MDNode* lscMetadata = m_inst->getMetadata("lsc.cache.ctrl"))
+                {
+                    newLI->setMetadata("lsc.cache.ctrl", lscMetadata);
+                }
+                return newLI;
             }
             else
             {