SWDEV-416218 - Set the image mapped flag correctly on COMGR path.

Change-Id: I6c0908731f10c8ff5b3e159c0cfb68ce096c7bdf
ROCm · Aug 10, 2023 · d18f4b1 · d18f4b1
1 parent 4969004
commit d18f4b1
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/hipamd/src/hip_fatbin.cpp b/hipamd/src/hip_fatbin.cpp
@@ -94,11 +94,13 @@ hipError_t FatBinaryInfo::ExtractFatBinaryUsingCOMGR(const std::vector<hip::Devi
       return hipErrorInvalidValue;
 
     // If image_ is nullptr, then file path is passed via hipMod* APIs, so map the file.
-    if (image_ == nullptr && !amd::Os::MemoryMapFileDesc(fdesc_, fsize_, foffset_, &image_)
-                          && (image_mapped_ = true)) {
-      LogError("Cannot map the file descriptor");
-      amd::Os::CloseFileHandle(fdesc_);
-      return hipErrorInvalidValue;
+    if (image_ == nullptr) {
+      if(!amd::Os::MemoryMapFileDesc(fdesc_, fsize_, foffset_, &image_)) {
+        LogError("Cannot map the file descriptor");
+        amd::Os::CloseFileHandle(fdesc_);
+        return hipErrorInvalidValue;
+      }
+      image_mapped_ = true;
     }
   }