From 30e9b1803cf19ffccc6d58d7168da869885a927d Mon Sep 17 00:00:00 2001 From: Matthias Diener <mdiener@illinois.edu> Date: Fri, 11 Jun 2021 15:25:23 -0500 Subject: [PATCH] fix previous --- arraycontext/__init__.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/arraycontext/__init__.py b/arraycontext/__init__.py index 5224666..b367382 100644 --- a/arraycontext/__init__.py +++ b/arraycontext/__init__.py @@ -53,7 +53,8 @@ from .container.traversal import ( from .impl.pyopencl import PyOpenCLArrayContext from .impl.pytato import PytatoArrayContext -from .pytest import pytest_generate_tests_for_array_contexts +from .pytest import (pytest_generate_tests_for_array_contexts, + pytest_generate_tests_for_pyopencl_array_context) from .loopy import make_loopy_program @@ -82,7 +83,8 @@ __all__ = ( "make_loopy_program", - "pytest_generate_tests_for_array_contexts" + "pytest_generate_tests_for_array_contexts", + "pytest_generate_tests_for_pyopencl_array_context" ) -- GitLab