[UR][L0] Make allocator tracing independent of setting the limits

Without this change, setting UR_L0_USM_ALLOCATOR_TRACE or
SYCL_PI_LEVEL_ZERO_USM_ALLOCATOR_TRACE only had an effect
if UR_L0_USM_ALLOCATOR or SYCL_PI_LEVEL_ZERO_USM_ALLOCATOR
was also set.

Author: igchor

source/adapters/level_zero/usm.cpp

@@ -49,14 +49,6 @@ ur_result_t umf2urResult(umf_result_t umfResult) {
 }
 
 usm::DisjointPoolAllConfigs InitializeDisjointPoolConfig() {
-  const char *PoolUrConfigVal = std::getenv("SYCL_PI_LEVEL_ZERO_USM_ALLOCATOR");
-  const char *PoolPiConfigVal = std::getenv("UR_L0_USM_ALLOCATOR");
-  const char *PoolConfigVal =
-      PoolUrConfigVal ? PoolUrConfigVal : PoolPiConfigVal;
-  if (PoolConfigVal == nullptr) {
-    return usm::DisjointPoolAllConfigs();
-  }
-
   const char *PoolUrTraceVal = std::getenv("UR_L0_USM_ALLOCATOR_TRACE");
   const char *PoolPiTraceVal =
       std::getenv("SYCL_PI_LEVEL_ZERO_USM_ALLOCATOR_TRACE");
@@ -69,6 +61,14 @@ usm::DisjointPoolAllConfigs InitializeDisjointPoolConfig() {
     PoolTrace = std::atoi(PoolTraceVal);
   }
 
+  const char *PoolUrConfigVal = std::getenv("SYCL_PI_LEVEL_ZERO_USM_ALLOCATOR");
+  const char *PoolPiConfigVal = std::getenv("UR_L0_USM_ALLOCATOR");
+  const char *PoolConfigVal =
+      PoolUrConfigVal ? PoolUrConfigVal : PoolPiConfigVal;
+  if (PoolConfigVal == nullptr) {
+    return usm::DisjointPoolAllConfigs(PoolTrace);
+  }
+
   return usm::parseDisjointPoolConfig(PoolConfigVal, PoolTrace);
 }
 

source/common/umf_pools/disjoint_pool_config_parser.cpp

@@ -24,7 +24,7 @@ constexpr auto operator""_GB(unsigned long long x) -> size_t {
     return x * 1024 * 1024 * 1024;
 }
 
-DisjointPoolAllConfigs::DisjointPoolAllConfigs() {
+DisjointPoolAllConfigs::DisjointPoolAllConfigs(int trace) {
     Configs[DisjointPoolMemType::Host].name = "Host";
     Configs[DisjointPoolMemType::Device].name = "Device";
     Configs[DisjointPoolMemType::Shared].name = "Shared";
@@ -58,6 +58,10 @@ DisjointPoolAllConfigs::DisjointPoolAllConfigs() {
     Configs[DisjointPoolMemType::SharedReadOnly].MaxPoolableSize = 4_MB;
     Configs[DisjointPoolMemType::SharedReadOnly].Capacity = 4;
     Configs[DisjointPoolMemType::SharedReadOnly].SlabMinSize = 2_MB;
+
+    for (auto &Config : Configs) {
+        Config.PoolTrace = trace;
+    }
 }
 
 DisjointPoolAllConfigs parseDisjointPoolConfig(const std::string &config,

source/common/umf_pools/disjoint_pool_config_parser.hpp

@@ -23,7 +23,7 @@ class DisjointPoolAllConfigs {
     size_t EnableBuffers = 1;
     DisjointPoolConfig Configs[DisjointPoolMemType::All];
 
-    DisjointPoolAllConfigs();
+    DisjointPoolAllConfigs(int trace = 0);
 };
 
 // Parse optional config parameters of this form: