From a59323a1ead23f296da47f314aff6116caa9e39a Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Thu, 2 Jan 2020 16:20:42 +0100
Subject: [PATCH] Avoid dependency on pycuda in crazy
---
src/pystencils_reco/_crazy_decorator.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/src/pystencils_reco/_crazy_decorator.py b/src/pystencils_reco/_crazy_decorator.py
index 93af028..f68bf38 100644
--- a/src/pystencils_reco/_crazy_decorator.py
+++ b/src/pystencils_reco/_crazy_decorator.py
@@ -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()}
--
GitLab