Fix for #158 (#159)

* Cache key * Update Project.toml Co-authored-by: Hong Ge <[email protected]>
TuringLang · Nov 2, 2022 · ca5f155 · ca5f155 · yebai · Nov 2, 2022
1 parent 5ec9f2b
commit ca5f155
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -3,7 +3,7 @@ uuid = "6f1fad26-d15e-5dc8-ae53-837a1d7b8c9f"
 license = "MIT"
 desc = "Tape based task copying in Turing"
 repo = "https://github.com/TuringLang/Libtask.jl.git"
-version = "0.8.3"
+version = "0.8.4"
 
 [deps]
 FunctionWrappers = "069b7b12-0de2-55c6-9aab-29f3d0a68a2e"

diff --git a/src/tapedfunction.jl b/src/tapedfunction.jl
@@ -62,7 +62,7 @@ mutable struct TapedFunction{F, TapeType}
 
     function TapedFunction{F, T}(f::F, args...; cache=false, deepcopy_types=Union{}) where {F, T}
         args_type = _accurate_typeof.(args)
-        cache_key = (f, args_type...)
+        cache_key = (f, deepcopy_types, args_type...)
 
         if cache && haskey(TRCache, cache_key) # use cache
             cached_tf = TRCache[cache_key]::TapedFunction{F, T}