Skip to content
Snippets Groups Projects
Commit a59323a1 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Avoid dependency on pycuda in crazy

parent 2b798ba4
No related merge requests found
......@@ -26,7 +26,6 @@ def crazy(function) -> pystencils_reco.AssignmentCollection:
# @disk_cache_no_fallback
@functools.wraps(function)
def wrapper(*args, **kwargs):
import pycuda.gpuarray # TODO(seitz): remove dependency
inspection = inspect.getfullargspec(function)
arg_names = inspection.args
annotations = inspection.annotations
......@@ -35,7 +34,7 @@ def crazy(function) -> pystencils_reco.AssignmentCollection:
if is_array_like(a)
else a for i, a in enumerate(args)}
compile_kwargs = {k: create_field_from_array_like(str(k), a, annotations.get(k, None))
if (hasattr(a, '__array__') or isinstance(a, pycuda.gpuarray.GPUArray)) and
if (hasattr(a, '__array__') or 'GPUArray' in str(a.__class__)) and
not isinstance(a, sympy.Matrix) # noqa
else a for (k, a) in kwargs.items()}
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment