From 633ce232d0674ceadda550855afb517504dc6973 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Tue, 29 Oct 2019 21:00:30 +0100
Subject: [PATCH] Fix create_cuda_kernel to allow CUSTOM_FIELDS to have a
 different size

---
 pystencils/gpucuda/kernelcreation.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pystencils/gpucuda/kernelcreation.py b/pystencils/gpucuda/kernelcreation.py
index 4430abba6..eecfe278e 100644
--- a/pystencils/gpucuda/kernelcreation.py
+++ b/pystencils/gpucuda/kernelcreation.py
@@ -20,7 +20,7 @@ def create_cuda_kernel(assignments,
     all_fields = fields_read.union(fields_written)
     read_only_fields = set([f.name for f in fields_read - fields_written])
 
-    buffers = set([f for f in all_fields if FieldType.is_buffer(f)])
+    buffers = set([f for f in all_fields if FieldType.is_buffer(f) or FieldType.is_custom(f)])
     fields_without_buffers = all_fields - buffers
 
     field_accesses = set()
-- 
GitLab