| 196 | constexpr size_t kDefaultThreadLocalCacheSize = 64; |
| 197 | |
| 198 | size_t GetTensorBufferPoolSize(size_t base = kDefaultPoolSizeBase) { |
| 199 | static double factor = |
| 200 | ParseFloatFromEnv("ONEFLOW_TENSOR_BUFFER_POOL_SIZE_FACTOR", kDefaultPoolSizeFactor); |
| 201 | return static_cast<size_t>(std::ceil(base * factor)); |
| 202 | } |
| 203 | |
| 204 | size_t GetTensorBufferPoolThreadLocalCacheSize() { |
| 205 | static size_t cache_size = ParseIntegerFromEnv( |
no test coverage detected