Address comment

teresajohnson · teresajohnson · commit d41836b2228f · 2024-10-20T22:45:56.000-07:00
diff --git a/llvm/lib/Transforms/IPO/MemProfContextDisambiguation.cpp b/llvm/lib/Transforms/IPO/MemProfContextDisambiguation.cpp
@@ -4432,6 +4432,8 @@ bool MemProfContextDisambiguation::applyImport(Module &M) {
       continue;
     for (auto &BB : F) {
       for (auto &I : BB) {
+        if (!isa<CallBase>(I))
+          continue;
         I.setMetadata(LLVMContext::MD_memprof, nullptr);
         I.setMetadata(LLVMContext::MD_callsite, nullptr);
       }

Original file line number	Diff line number	Diff line change
`@@ -4432,6 +4432,8 @@ bool MemProfContextDisambiguation::applyImport(Module &M) {`
`4432`	`4432`	`continue;`
`4433`	`4433`	`for (auto &BB : F) {`
`4434`	`4434`	`for (auto &I : BB) {`
	`4435`	`+ if (!isa<CallBase>(I))`
	`4436`	`+ continue;`
`4435`	`4437`	`I.setMetadata(LLVMContext::MD_memprof, nullptr);`
`4436`	`4438`	`I.setMetadata(LLVMContext::MD_callsite, nullptr);`
`4437`	`4439`	`}`