Skip to content

Commit

Permalink
Update
Browse files Browse the repository at this point in the history
  • Loading branch information
pxl-th committed Jan 7, 2025
1 parent a1cec2e commit 3139c0c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/cache_allocator.jl
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
const CuCacheAllocator = GPUArrays.AllocCache.PerDeviceCacheAllocator(CuArray; free_immediately=true)
const CuCacheAllocator = GPUArrays.AllocCache.PerDeviceCacheAllocator(CuArray)

GPUArrays.AllocCache.cache_allocator(::Type{<: CuArray}) = CuCacheAllocator

Expand Down

0 comments on commit 3139c0c

Please sign in to comment.