diff --git a/src/pystencils_autodiff/backends/_pytorch.py b/src/pystencils_autodiff/backends/_pytorch.py index 6d2da60f799483a5dbc62f0266d6d331ee750cb6..22242763a6830c571af6d155faf6c2f76e29e598 100644 --- a/src/pystencils_autodiff/backends/_pytorch.py +++ b/src/pystencils_autodiff/backends/_pytorch.py @@ -100,7 +100,9 @@ def tensor_to_gpuarray(tensor): raise ValueError( 'Cannot convert CPU tensor to GPUArray (call `cuda()` on it)') else: - return pycuda.gpuarray.GPUArray(tensor.shape, dtype=torch_dtype_to_numpy(tensor.dtype), gpudata=tensor.data_ptr()) + return pycuda.gpuarray.GPUArray(tensor.shape, + dtype=torch_dtype_to_numpy(tensor.dtype), + gpudata=tensor.data_ptr()) def gpuarray_to_tensor(gpuarray, context=None): diff --git a/src/pystencils_autodiff/backends/_tensorflow_cpp.py b/src/pystencils_autodiff/backends/_tensorflow_cpp.py index 541e1225c61eefd0ee095c035dbcf59d4b51598f..d64f1765bc4756e84ddaa5da0cf2eb5cac7ea7bf 100644 --- a/src/pystencils_autodiff/backends/_tensorflow_cpp.py +++ b/src/pystencils_autodiff/backends/_tensorflow_cpp.py @@ -6,7 +6,8 @@ Advantages: - Custom Ops will be serializable Disadavantages: -- C++ Code has to be build with correct parameters and ABI for present Tensorflow version (best integrated into Tensorflow build) +- C++ Code has to be build with correct parameters and ABI +for present Tensorflow version (best integrated into Tensorflow build) """ diff --git a/src/pystencils_autodiff/lbm/__init__.py b/src/pystencils_autodiff/lbm/__init__.py index b0579476c1af869e4c25cc3b9582c623bb2b4115..a28beebcc67532ac1db816dfc49497edd4a0e301 100644 --- a/src/pystencils_autodiff/lbm/__init__.py +++ b/src/pystencils_autodiff/lbm/__init__.py @@ -1,2 +1,4 @@ from lbmpy.autodiff._autodiff_lbstep import AutoDiffLatticeBoltzmannStep from lbmpy.autodiff.adjoint_boundaryconditions import AdjointBoundaryCondition + +__all__ = ['AutoDiffLatticeBoltzmannStep', 'AdjointBoundaryCondition'] diff --git a/src/pystencils_autodiff/tensorflow.py b/src/pystencils_autodiff/tensorflow.py index 82a86cca85713dbee6ef2d9c8336a647aa99634c..91a8c76436ec4fd100b8bb72016d6e05832cef5f 100644 --- a/src/pystencils_autodiff/tensorflow.py +++ b/src/pystencils_autodiff/tensorflow.py @@ -24,5 +24,3 @@ def tf_variable_from_field(field, init_val=0, constraint=None): def tf_placeholder_from_field(field): return tf.placeholder(dtype=field.dtype.numpy_dtype, name=field.name + '_placeholder', shape=field.shape) - -