From 185aeafbfd58eaaca91ce406ac00bc12f7985008 Mon Sep 17 00:00:00 2001
From: Matthias Diener <mdiener@illinois.edu>
Date: Thu, 7 Apr 2022 14:32:07 -0500
Subject: [PATCH] add PooledBuffer to dedup cache key (#304)

* add PooledBuffer to dedup cache key

* fix mypy

* change to MemoryObjectHolder

* workaround for doc build
---
 doc/conf.py         | 2 +-
 pytato/transform.py | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/doc/conf.py b/doc/conf.py
index 2991a2f..74912f3 100644
--- a/doc/conf.py
+++ b/doc/conf.py
@@ -35,5 +35,5 @@ sys.PYTATO_BUILDING_SPHINX_DOCS = True
 nitpick_ignore_regex = [
     ["py:class", r"numpy.(u?)int[\d]+"],
     ["py:class", r"pyrsistent.typing.(.+)"],
-
+    ["py:class", r"typing_extensions(.+)"],
 ]
diff --git a/pytato/transform.py b/pytato/transform.py
index d16c865..04a19f7 100644
--- a/pytato/transform.py
+++ b/pytato/transform.py
@@ -1716,10 +1716,11 @@ def _get_data_dedup_cache_key(ary: DataInterface) -> Hashable:
     import sys
     if "pyopencl" in sys.modules:
         from pyopencl.array import Array as CLArray  # type: ignore[import]
-        from pyopencl import Buffer as CLBuffer
+        from pyopencl import MemoryObjectHolder
         if isinstance(ary, CLArray):
             # pyopencl represents 0-long arrays' base_data as None
-            assert isinstance(ary.base_data, CLBuffer) or (ary.base_data is None)
+            assert (isinstance(ary.base_data, MemoryObjectHolder)
+                    or (ary.base_data is None))
             return (
                     None if ary.base_data is None else ary.base_data.int_ptr,
                     ary.offset,
-- 
GitLab