diff --git a/src/pystencils_autodiff/__init__.py b/src/pystencils_autodiff/__init__.py index 330bed690611f22fa19f55d8d564f0158ef48d08..e15b7d2fd9ece8e4801e43fc85362a18684e4eb6 100644 --- a/src/pystencils_autodiff/__init__.py +++ b/src/pystencils_autodiff/__init__.py @@ -20,4 +20,3 @@ __all__ = ['backends', sys.modules['pystencils.autodiff'] = pystencils_autodiff sys.modules['pystencils.autodiff.backends'] = pystencils_autodiff.backends - diff --git a/src/pystencils_autodiff/framework_integration/astnodes.py b/src/pystencils_autodiff/framework_integration/astnodes.py index fdfe10064206b47ce28685796a7e94ae81793f00..e7fc9139c1c9e8986928607ce23fe24210f22b0a 100644 --- a/src/pystencils_autodiff/framework_integration/astnodes.py +++ b/src/pystencils_autodiff/framework_integration/astnodes.py @@ -154,7 +154,7 @@ auto {texture_object}Destroyer = [&](){{ device_ptr=self._device_ptr, cuda_channel_format=self._get_channel_format_string(), bits_per_channel=self._dtype.itemsize * 8, - total_size=self._dtype.itemsize * reduce(lambda x, y: x*y, self._shape, 1)) + total_size=self._dtype.itemsize * reduce(lambda x, y: x * y, self._shape, 1)) return code