commit bbdc5b5c07ea72cbaf428c92c816678516c49946
author Shilei Tian <tianshilei1992@gmail.com> Mon Oct 25 17:15:56 2021 -0400
committer Copybara-Service <copybara-worker@google.com> Mon Oct 25 14:17:32 2021 -0700
tree a0c801d22786459f6c04fdd4be1e118e656d2c5a
parent 9b5027ff1df6cf2e963ba46afef2d087363cb9de

[OpenMP][Offloading] Only get trip count if team construct

Reviewed By: grokos

Differential Revision: https://reviews.llvm.org/D112475

GitOrigin-RevId: 2a30c03c62070d12ce6729441c2c58360dcc1024

libomptarget/src/omptarget.cpp

diff --git a/libomptarget/src/omptarget.cpp b/libomptarget/src/omptarget.cpp
index 5ec0609..06370ac 100644
--- a/libomptarget/src/omptarget.cpp
+++ b/libomptarget/src/omptarget.cpp
@@ -1441,9 +1441,6 @@
     }
   }
 
-  // Get loop trip count
-  uint64_t LoopTripCount = getLoopTripCount(DeviceId);
-
   // Launch device execution.
   void *TgtEntryPtr = TargetTable->EntriesBegin[TM->Index].addr;
   DP("Launching target execution %s with pointer " DPxMOD " (index=%d).\n",
@@ -1455,7 +1452,7 @@
     if (IsTeamConstruct)
       Ret = Device.runTeamRegion(TgtEntryPtr, &TgtArgs[0], &TgtOffsets[0],
                                  TgtArgs.size(), TeamNum, ThreadLimit,
-                                 LoopTripCount, AsyncInfo);
+                                 getLoopTripCount(DeviceId), AsyncInfo);
     else
       Ret = Device.runRegion(TgtEntryPtr, &TgtArgs[0], &TgtOffsets[0],
                              TgtArgs.size(), AsyncInfo);