Skip to content

Commit e39860a

Browse files
authored
Update comment about memory pool.
Confirmed with NVIDIA that his is how it works. [skip ci]
1 parent 9796d5a commit e39860a

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

src/pool.jl

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -176,9 +176,8 @@ function pool_mark(dev::CuDevice)
176176
if status[] === nothing
177177
limits = memory_limits()
178178

179-
# configure our memory pool
180-
# XXX: is it OK to replace the pool like this? we need to, for setting limits.
181-
# how do we get other applications/libraries to use it?
179+
# create a custom memory pool and assign it to the device
180+
# so that other libraries and applications will use it.
182181
pool = if limits.hard > 0 && CUDA.driver_version() >= v"12.2"
183182
CuMemoryPool(dev; maxSize=limits.hard)
184183
else

0 commit comments

Comments
 (0)