[llvm] [Offload] Make MemoryManager threshold ENV var size_t type. (PR #124063)

[via llvm-commits]

https://github.com/hidekisaito created https://github.com/llvm/llvm-project/pull/124063

None

>From 551aa2c50b1550e9b8a36e078dfb05b8410a6839 Mon Sep 17 00:00:00 2001
From: Hideki Saito <hidekido at amd.com>
Date: Wed, 22 Jan 2025 22:31:13 -0600
Subject: [PATCH] [Offload] Make MemoryManager threshold ENV var size_t type.

---
 offload/include/Shared/EnvironmentVar.h                | 1 +
 offload/plugins-nextgen/common/include/MemoryManager.h | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/offload/include/Shared/EnvironmentVar.h b/offload/include/Shared/EnvironmentVar.h
index 82f434e91a85b8..c1854774d572b6 100644
--- a/offload/include/Shared/EnvironmentVar.h
+++ b/offload/include/Shared/EnvironmentVar.h
@@ -128,6 +128,7 @@ using Int32Envar = Envar<int32_t>;
 using Int64Envar = Envar<int64_t>;
 using UInt32Envar = Envar<uint32_t>;
 using UInt64Envar = Envar<uint64_t>;
+using SizeTEnvar = Envar<size_t>;
 using StringEnvar = Envar<std::string>;
 using BoolEnvar = Envar<bool>;
 
diff --git a/offload/plugins-nextgen/common/include/MemoryManager.h b/offload/plugins-nextgen/common/include/MemoryManager.h
index fe1989930b76ef..34e9e9618680f3 100644
--- a/offload/plugins-nextgen/common/include/MemoryManager.h
+++ b/offload/plugins-nextgen/common/include/MemoryManager.h
@@ -324,7 +324,7 @@ class MemoryManagerTy {
   /// manager explicitly by setting the var to 0. If user doesn't specify
   /// anything, returns <0, true>.
   static std::pair<size_t, bool> getSizeThresholdFromEnv() {
-    static UInt32Envar MemoryManagerThreshold(
+    static SizeTEnvar MemoryManagerThreshold(
         "LIBOMPTARGET_MEMORY_MANAGER_THRESHOLD", 0);
 
     size_t Threshold = MemoryManagerThreshold.get();