From 3914993a88b7cfd0451c7ddbb8cf50e3da64b9d9 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Fri, 28 Feb 2020 15:08:34 +0100
Subject: [PATCH] Avoid making AdjointField custom if forward field CUSTOM
---
src/pystencils_autodiff/_adjoint_field.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/src/pystencils_autodiff/_adjoint_field.py b/src/pystencils_autodiff/_adjoint_field.py
index c964f65..b76d2d4 100644
--- a/src/pystencils_autodiff/_adjoint_field.py
+++ b/src/pystencils_autodiff/_adjoint_field.py
@@ -11,7 +11,9 @@ class AdjointField(pystencils.Field):
def __init__(self, forward_field, name_prefix='diff'):
new_name = name_prefix + forward_field.name
- super().__init__(new_name, forward_field.field_type, forward_field._dtype,
+ super().__init__(new_name, pystencils.FieldType.GENERIC
+ if forward_field.field_type != pystencils.FieldType.BUFFER
+ else pystencils.FieldType.BUFFER, forward_field._dtype,
forward_field._layout, forward_field.shape, forward_field.strides)
self.corresponding_forward_field = forward_field
self.name_prefix = name_prefix
--
GitLab