feat: fix module AST cache

Author: WeiTaShao

pykokkos/core/compiler.py

@@ -181,7 +181,9 @@ def compile_object(
             # Avoid fusing the ASTs before checking if it was already compiled
             entity, classtypes = self.fuse_objects(metadata, fuse_ASTs=False, **kwargs)
 
-        hash: str = self.members_hash(entity.path, entity.name, types_signature)
+        hash: str = self.members_hash(
+            entity.path, entity.name, module_setup.ast_signature, types_signature
+        )
 
         types_inferred: bool = updated_types is not None
         decorator_inferred: bool = updated_decorator is not None
@@ -413,21 +415,26 @@ def read_defaults(self) -> Optional[CompilationDefaults]:
         return defaults
 
     def members_hash(
-        self, path: List[str], name: str, types_signature: Optional[str]
+        self,
+        path: List[str],
+        name: str,
+        ast_signature: str,
+        types_signature: Optional[str],
     ) -> str:
         """
         Map from entity path and name to a string to index members
 
         :param path: the path to the file containing the entity
         :param name: the name of the entity
+        :param ast_signature: signature of the translated AST and dependencies
         :param types_signature: string signature of inferred parameter types
         :returns: the hash of the entity
         """
 
         return (
-            f"{path}_{name}"
+            f"{path}_{name}_{ast_signature}"
             if types_signature is None
-            else f"{path}_{name}_{types_signature}"
+            else f"{path}_{name}_{ast_signature}_{types_signature}"
         )
 
     def extract_members(