modify default value

DeepLink-org · Apr 8, 2024 · ff63f12 · ff63f12
1 parent ea2db53
commit ff63f12
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dipu/torch_dipu/csrc_dipu/runtime/core/allocator/DIPUBFCachingAllocator.cpp b/dipu/torch_dipu/csrc_dipu/runtime/core/allocator/DIPUBFCachingAllocator.cpp
@@ -14,7 +14,7 @@ namespace dipu {
 
 // NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)
 const size_t kMaxExtendSize = []() {
-  size_t maxExtendSize = 512;
+  size_t maxExtendSize = 256;
   const char* env = std::getenv("DIPU_MAX_EXTEND_SIZE");
   if (env != nullptr) {
     maxExtendSize = std::atoi(env);

diff --git a/dipu/torch_dipu/csrc_dipu/runtime/core/allocator/DIPUCachingAllocator.cpp b/dipu/torch_dipu/csrc_dipu/runtime/core/allocator/DIPUCachingAllocator.cpp
@@ -20,7 +20,7 @@ std::mutex DIPURawDeviceAllocator::mutex_;
 
 // NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)
 size_t kMaxAsyncResourcePoolLength = []() {
-  size_t maxAsyncResourcePoolLength = 32;
+  size_t maxAsyncResourcePoolLength = 8;
   const char* env = std::getenv("DIPU_MAX_ASYNC_RESOURCE_POOL_LENGTH");
   if (env != nullptr) {
     maxAsyncResourcePoolLength = std::atoi(env);