[OpenMP] Mute OpenMP Target Enter, Exit and Data codegen for device pass

[TIFitis]

No description provided.

[llvmbot]

@llvm/pr-subscribers-flang-openmp

Author: Akash Banerjee (TIFitis)

Changes

---

Full diff: https://github.com/llvm/llvm-project/pull/72287.diff

1 Files Affected:

• (modified) llvm/lib/Frontend/OpenMP/OMPIRBuilder.cpp (+4)

diff --git a/llvm/lib/Frontend/OpenMP/OMPIRBuilder.cpp b/llvm/lib/Frontend/OpenMP/OMPIRBuilder.cpp
index 24d15267a65e933..4e25d57a06a4f81 100644
--- a/llvm/lib/Frontend/OpenMP/OMPIRBuilder.cpp
+++ b/llvm/lib/Frontend/OpenMP/OMPIRBuilder.cpp
@@ -4566,6 +4566,10 @@ OpenMPIRBuilder::InsertPointTy OpenMPIRBuilder::createTargetData(
   if (!updateToLocation(Loc))
     return InsertPointTy();
 
+  // Disable TargetData CodeGen on Device pass.
+  if (Config.isTargetDevice())
+    return Builder.saveIP();
+
   Builder.restoreIP(CodeGenIP);
   bool IsStandAlone = !BodyGenCB;
   MapInfosTy *MapInfo;

[agozillon]

Thank you Akash, it LGTM. Might be worth a second opinion from @jsjodin