CL Array, function

inducer · Jan 25, 2024 · cd167af · cd167af
1 parent 8f03fa5
commit cd167af
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pytato/analysis/__init__.py b/pytato/analysis/__init__.py
@@ -478,6 +478,13 @@ def update_for_ndarray(self, key_hash: Any, key: Any) -> None:
     def update_for_TaggableCLArray(self, key_hash: Any, key: Any) -> None:
         self.update_for_ndarray(key_hash, key.get())
 
+    def update_for_Array(self, key_hash: Any, key: Any) -> None:
+        # CL Array
+        self.update_for_ndarray(key_hash, key.get())
+
+    def update_for_function(self, key_hash: Any, key: Any) -> None:
+        self.rec(key_hash, key.__name__)
+
     def update_for_pymbolic_expression(self, key_hash: Any, key: Any) -> None:
         if key is None:
             self.update_for_NoneType(key_hash, key)