From 5854a211eb916d7bff372a364739439c08aacdfd Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Thu, 14 Mar 2019 22:10:57 -0500
Subject: [PATCH] Fix test_mempool_2

---
 test/test_wrapper.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/test_wrapper.py b/test/test_wrapper.py
index f7fa55ee..ea50bd5a 100644
--- a/test/test_wrapper.py
+++ b/test/test_wrapper.py
@@ -552,7 +552,7 @@ def test_mempool(ctx_factory):
 
 
 def test_mempool_2(ctx_factory):
-    from pyopencl.tools import MemoryPool
+    from pyopencl.tools import MemoryPool, ImmediateAllocator
     from random import randrange
 
     context = ctx_factory()
-- 
GitLab