File tree Expand file tree Collapse file tree 1 file changed +14
-12
lines changed
Expand file tree Collapse file tree 1 file changed +14
-12
lines changed Original file line number Diff line number Diff line change @@ -219,18 +219,20 @@ def get_target(self):
219219
220220# {{{ PytatoPyOpenCLArrayContext
221221
222- from pytato .target .loopy import LoopyPyOpenCLTarget
223-
224-
225- class _ArgSizeLimitingPytatoLoopyPyOpenCLTarget (LoopyPyOpenCLTarget ):
226- def __init__ (self , limit_arg_size_nbytes : int ) -> None :
227- super ().__init__ ()
228- self .limit_arg_size_nbytes = limit_arg_size_nbytes
229-
230- @memoize_method
231- def get_loopy_target (self ) -> Optional ["lp.PyOpenCLTarget" ]:
232- from loopy import PyOpenCLTarget
233- return PyOpenCLTarget (limit_arg_size_nbytes = self .limit_arg_size_nbytes )
222+ try :
223+ from pytato .target .loopy import LoopyPyOpenCLTarget
224+
225+ class _ArgSizeLimitingPytatoLoopyPyOpenCLTarget (LoopyPyOpenCLTarget ):
226+ def __init__ (self , limit_arg_size_nbytes : int ) -> None :
227+ super ().__init__ ()
228+ self .limit_arg_size_nbytes = limit_arg_size_nbytes
229+
230+ @memoize_method
231+ def get_loopy_target (self ) -> Optional ["lp.PyOpenCLTarget" ]:
232+ from loopy import PyOpenCLTarget
233+ return PyOpenCLTarget (limit_arg_size_nbytes = self .limit_arg_size_nbytes )
234+ except ImportError :
235+ pass
234236
235237
236238class PytatoPyOpenCLArrayContext (_BasePytatoArrayContext ):
You can’t perform that action at this time.
0 commit comments