Skip to content

Commit 57f8a44

Browse files
authored
[SYCL] Change USM pooling parameters. (#5457)
1 parent 8213321 commit 57f8a44

File tree

1 file changed

+9
-10
lines changed

1 file changed

+9
-10
lines changed

sycl/plugins/level_zero/usm_allocator.cpp

Lines changed: 9 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -106,18 +106,18 @@ static class SetLimits {
106106
size_t SlabMinSize[3] = {64 * 1024, 64 * 1024, 2 * 1024 * 1024};
107107

108108
// Allocations up to this limit will be subject to chunking/pooling
109-
size_t MaxPoolableSize[3] = {0, 32 * 1024, 0};
109+
size_t MaxPoolableSize[3] = {2 * 1024 * 1024, 4 * 1024 * 1024, 0};
110110

111111
// When pooling, each bucket will hold a max of 4 unfreed slabs
112-
size_t Capacity[3] = {0, 0, 0};
112+
size_t Capacity[3] = {4, 4, 0};
113113

114114
// Maximum memory left unfreed in pool
115-
size_t MaxPoolSize = 0;
115+
size_t MaxPoolSize = 16 * 1024 * 1024;
116116

117117
size_t CurPoolSize = 0;
118118
size_t CurPoolSizes[3] = {0, 0, 0};
119119

120-
bool EnableBuffers = false;
120+
bool EnableBuffers = true;
121121

122122
// Whether to print pool usage statistics
123123
int PoolTrace = 0;
@@ -134,17 +134,16 @@ static class SetLimits {
134134
// pool size for all contexts.
135135
// Duplicate specifications will result in the right-most taking effect.
136136
//
137-
// Current defaults are to match pre-2021.3 pooling.
138137
// EnableBuffers: Apply chunking/pooling to SYCL buffers.
139-
// Default 0 (false).
138+
// Default true.
140139
// MaxPoolSize: Limit on overall unfreed memory.
141-
// Default 0MB.
140+
// Default 16MB.
142141
// MaxPoolableSize: Maximum allocation size subject to chunking/pooling.
143-
// Default 32KB.
142+
// Default 2MB host, 4MB device and 0 shared.
144143
// Capacity: Maximum number of unfreed allocations in each bucket.
145-
// Default 0.
144+
// Default 4.
146145
// SlabMinSize: Minimum allocation size requested from USM.
147-
// Default 64KB.
146+
// Default 64KB host and device, 2MB shared.
148147
//
149148
// Example of usage:
150149
// SYCL_PI_LEVEL_ZERO_USM_ALLOCATOR=1;32M;host:1M,4,64K;device:1M,4,64K;shared:0,0,2M

0 commit comments

Comments
 (0)