CL Array, function

inducer · Nov 7, 2023 · b31657f · b31657f
1 parent a6a8782
commit b31657f
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pytato/analysis/__init__.py b/pytato/analysis/__init__.py
@@ -468,6 +468,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)