fix cugc

AStupidBear · Mar 8, 2020 · 96b24db · 96b24db · AStupidBear · Mar 8, 2020
1 parent 9e4f540
commit 96b24db
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "FluxUtils"
 uuid = "479fec60-eb0f-11e8-3e82-6d728fa6b650"
 authors = ["Yao Lu <[email protected]>"]
-version = "0.1.4"
+version = "0.1.5"
 
 [deps]
 BSON = "fbb218c0-5317-5bc6-957e-2ee96dd4b1f0"

diff --git a/src/util.jl b/src/util.jl
@@ -3,7 +3,7 @@ export notrack, cugc
 function cugc()
     GC.gc()
     isdefined(Flux, :CuArrays) &&
-    Flux.CuArrays.reclaim(true)
+    Flux.CuArrays.BinnedPool.reclaim()
 end
 
 function notrack(m)