Merge branch 'lwawrzyniak/fix-generic-overload-caching' into 'main'

Fix generic function caching See merge request omniverse/warp!573
NVIDIA · Jun 14, 2024 · bdb9e2a · bdb9e2a
2 parents cff9f2c + 916e251
commit bdb9e2a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -3,6 +3,7 @@
 ## [Upcoming Release] - 2024-??-??
 
 - Improve memory usage and performance for rigid body contact handling when `self.rigid_mesh_contact_max` is zero (default behavior)
+- Fix generic function caching
 
 ## [1.2.0] - 2024-06-06
 

diff --git a/warp/context.py b/warp/context.py
@@ -1488,8 +1488,9 @@ def hash_recursive(module, visited):
 
                 # functions source
                 for function in module.functions.values():
-                    # include all overloads
-                    for sig, func in function.user_overloads.items():
+                    # include all concrete and generic overloads
+                    overloads = itertools.chain(function.user_overloads.items(), function.user_templates.items())
+                    for sig, func in overloads:
                         # signature
                         ch.update(bytes(sig, "utf-8"))