| 25 | static constexpr size_t kDefaultTensorBufferAlignedSize = 1024; |
| 26 | |
| 27 | size_t GetTensorBufferAlignedSize(size_t origin_size, double factor) { |
| 28 | static size_t aligned_size = |
| 29 | ParseIntegerFromEnv("ONEFLOW_TENSOR_BUFFER_ALIGNED_SIZE", kDefaultTensorBufferAlignedSize); |
| 30 | return RoundUp(static_cast<size_t>(origin_size * factor), aligned_size); |
| 31 | } |
| 32 | |
| 33 | size_t GetTensorBufferGrowthSize(size_t origin_size) { |
| 34 | static double factor = |
no test coverage detected