From b31657fddf3ab4f2543cc437ba9e85eddb4f808b Mon Sep 17 00:00:00 2001 From: Matthias Diener Date: Thu, 28 Sep 2023 18:24:55 -0500 Subject: [PATCH] CL Array, function --- pytato/analysis/__init__.py | 7 +++++++ 1 file changed, 7 insertions(+) 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)