From de26e24ec99e2afebf388a2959100ff2e1e1cd6d Mon Sep 17 00:00:00 2001
From: Matthias Diener <mdiener@illinois.edu>
Date: Fri, 25 Jun 2021 18:52:14 -0500
Subject: [PATCH] fix queue
---
arraycontext/pytest.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/arraycontext/pytest.py b/arraycontext/pytest.py
index db5dcb0..77e51fe 100644
--- a/arraycontext/pytest.py
+++ b/arraycontext/pytest.py
@@ -100,8 +100,9 @@ class _PytatoPyOpenCLArrayContextFactory(PytestPyOpenCLArrayContextFactory):
force_device_scalars = False
def __call__(self):
- from arraycontext.impl.pytato import PytatoPyOpenCLArrayContext
- return PytatoPyOpenCLArrayContext(self.get_command_queue())
+ from arraycontext import PytatoPyOpenCLArrayContext
+ ctx, queue = self.get_command_queue()
+ return PytatoPyOpenCLArrayContext(queue)
def __str__(self):
return ("<Pytato array context factory for <pyopencl.Device '%s' on '%s'>"
--
GitLab