From 4a87eaf5c4a8207eb1f3a7a908028ece81d3ab78 Mon Sep 17 00:00:00 2001
From: Matthias Diener <mdiener@illinois.edu>
Date: Thu, 22 Sep 2022 16:58:02 -0500
Subject: [PATCH] assert alloc type
---
test/test_pytato_arraycontext.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/test/test_pytato_arraycontext.py b/test/test_pytato_arraycontext.py
index d13b1ed..fd3c6ea 100644
--- a/test/test_pytato_arraycontext.py
+++ b/test/test_pytato_arraycontext.py
@@ -175,6 +175,11 @@ def test_pytato_actx_allocator(actx_factory, pass_allocator):
return_value=False):
actx = _PytatoPyOpenCLArrayContextForTests(base_actx.queue,
allocator=alloc, use_memory_pool=use_memory_pool)
+
+ from pyopencl.tools import ImmediateAllocator, MemoryPool
+ assert isinstance(actx.allocator,
+ MemoryPool if use_memory_pool else ImmediateAllocator)
+
f = actx.compile(twice)
res = actx.to_numpy(f(99))
--
GitLab