diff --git a/pytato/analysis/__init__.py b/pytato/analysis/__init__.py index 8b2af0840..91d904e70 100644 --- 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)