refactor(dipu): move some env vars to environs.hpp

dipu/torch_dipu/csrc_dipu/base/environ.hpp

@@ -1,4 +1,4 @@
 // Copyright (c) 2024, DeepLink.
 //
 // dipu::environ contains all DIPU configurations set via environment variables.
 //
@@ -86,11 +86,27 @@
 // applyDelayedRegister() is called.
 DIPU_ENV_VAR(immediateRegisterOp, "DIPU_IMMEDIATE_REGISTER_OP", bool, false);
 inline const std::string kTorchAllocatorName = "TORCH";
+// Determine the name of the host memory cache algorithm 
+// based on the current environment configuration.
 DIPU_ENV_VAR(hostMemCachingAlgorithm, "DIPU_HOST_MEMCACHING_ALGORITHM",
              std::string, kTorchAllocatorName);
+// Used to specify the name of the device memory cache algorithm.
 DIPU_ENV_VAR(deviceMemCachingAlgorithm, "DIPU_DEVICE_MEMCACHING_ALGORITHM",
              std::string, kTorchAllocatorName);
+// Used to configure and initialize an instance of an object "CachingAllocatorConfig".
 DIPU_ENV_VAR(torchAllocatorConf, "DIPU_TORCH_ALLOCATOR_CONF", std::string, "");
+// maxExtendSize is used to limit the maximum size of an extension 
+// in the memory allocation in function of "extend()".
+DIPU_ENV_VAR(maxExtendSize, "DIPU_MAX_EXTEND_SIZE", std::size_t, 1024);
+// Configure a value to limit the maximum length of the asynchronous resource pool 
+// to avoid resource leakage and optimize resource management.
+inline const std::size_t kDefaultMaxAsyncResourcePoolLength = 96;
+DIPU_ENV_VAR(maxAsyncResourcePoolLength, "DIPU_MAX_ASYNC_RESOURCE_POOL_LENGTH", std::size_t, kDefaultMaxAsyncResourcePoolLength);
+// Control whether to force the use of back-off mode for P2P copy operation between Ascend chips.
+DIPU_ENV_VAR(forceFallbackP2pCopybetweenascends, "DIPU_FORCE_FALLBACK_ASCEND_P2P_COPY", bool, false);
+// Configure a numerical value to control the device 's affinity settings 
+// on the CPU to optimize thread scheduling during concurrent execution.
+DIPU_ENV_VAR(affinityCpuAffinit, "DIPU_CPU_AFFINITY", std::size_t, 0);
 
 #undef DIPU_ENV_VAR
 

dipu/torch_dipu/csrc_dipu/runtime/core/allocator/DIPUBFCachingAllocator.cpp

@@ -1,4 +1,4 @@
 // Copyright (c) 2023, DeepLink.
 
 #include <functional>
 #include <memory>
@@ -12,11 +12,12 @@
 #include "DIPUCachingAllocator.h"
 #include "DIPUSpinMutex.h"
 
+#include "csrc_dipu/base/environ.hpp"
+
 namespace dipu {
 
 // NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)
-const size_t kMaxExtendSize = get_env_or_default("DIPU_MAX_EXTEND_SIZE", 1024)
-                              << 20U;
+const size_t kMaxExtendSize = environ::maxExtendSize() << 20U;
 
 class BFCachingAllocatorImpl {
  public:

dipu/torch_dipu/csrc_dipu/runtime/core/allocator/DIPUCachingAllocator.cpp

@@ -1,4 +1,4 @@
 // Copyright (c) 2023, DeepLink.
 
 #include "DIPUCachingAllocator.h"
 
@@ -20,15 +20,15 @@
 #include "DIPUCachingDeviceAllocator.h"
 #include "DIPUCachingHostAllocator.h"
 
+#include "csrc_dipu/base/environ.hpp"
+
 namespace dipu {
 
 // NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)
 std::mutex DIPURawDeviceAllocator::mutex_;
 
-constexpr size_t kDefaultMaxAsyncResourcePoolLength = 96;
 // NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)
-const size_t kMaxAsyncResourcePoolLength = get_env_or_default(
-    "DIPU_MAX_ASYNC_RESOURCE_POOL_LENGTH", kDefaultMaxAsyncResourcePoolLength);
+const size_t kMaxAsyncResourcePoolLength = environ::maxAsyncResourcePoolLength();
 
 namespace {
 

dipu/torch_dipu/csrc_dipu/runtime/devproxy/deviceproxy.cpp

@@ -64,7 +64,8 @@ deviceId_t current_device() {
 }
 
 void setCpuAffinity(const int device) {
-  static int affinity = get_env_or_default("DIPU_CPU_AFFINITY", 0);
+  static int affinity = environ::affinityCpuAffinit();
+
   if (affinity < 0) {
     return;
   }

dipu/torch_dipu/csrc_dipu/vendor/ascend/deviceimpl.cpp

@@ -1,4 +1,4 @@
 // Copyright (c) 2023, DeepLink.
 #include <acl/acl.h>
 #include <acl/acl_op.h>
 #include <acl/acl_op_compiler.h>
@@ -18,6 +18,8 @@
 
 #include "basecommimpl.hpp"
 
+#include "csrc_dipu/base/environ.hpp"
+
 namespace dipu {
 
 namespace devapis {
@@ -30,8 +32,7 @@
 
 namespace {
 
-const bool forceFallbackP2PCopy =
-    get_env_or_default("DIPU_FORCE_FALLBACK_ASCEND_P2P_COPY", false);
+const bool forceFallbackP2PCopy = environ::forceFallbackP2pCopybetweenascends();
 
 class NpuP2PInfo {
   enum class P2pStatus : int8_t {