Skip to content
Snippets Groups Projects

Use int64 for indexing

Merged Markus Holzer requested to merge holzer/pystencils:type_boundary into master
3 files
+ 2
3
Compare changes
  • Side-by-side
  • Inline
Files
3
@@ -444,7 +444,6 @@ class BoundaryOffsetInfo(CustomCodeNode):
@@ -444,7 +444,6 @@ class BoundaryOffsetInfo(CustomCodeNode):
def create_boundary_kernel(field, index_field, stencil, boundary_functor, target='cpu', **kernel_creation_args):
def create_boundary_kernel(field, index_field, stencil, boundary_functor, target='cpu', **kernel_creation_args):
elements = [BoundaryOffsetInfo(stencil)]
elements = [BoundaryOffsetInfo(stencil)]
index_arr_dtype = index_field.dtype.numpy_dtype
dir_symbol = TypedSymbol("dir", np.int64)
dir_symbol = TypedSymbol("dir", np.int64)
elements += [Assignment(dir_symbol, index_field[0]('dir'))]
elements += [Assignment(dir_symbol, index_field[0]('dir'))]
elements += boundary_functor(field, direction_symbol=dir_symbol, index_field=index_field)
elements += boundary_functor(field, direction_symbol=dir_symbol, index_field=index_field)
Loading