Skip to content
Snippets Groups Projects
Commit 15fc443b authored by Markus Holzer's avatar Markus Holzer
Browse files

small adaption for consistency

parent 0ff73c3e
No related branches found
No related tags found
1 merge request!84Use int64 for indexing
...@@ -219,7 +219,7 @@ class NeighbourOffsetArrays(CustomCodeNode): ...@@ -219,7 +219,7 @@ class NeighbourOffsetArrays(CustomCodeNode):
def _offset_symbols(dim): def _offset_symbols(dim):
return [TypedSymbol(f"neighbour_offset_{d}", create_type(np.int64)) for d in ['x', 'y', 'z'][:dim]] return [TypedSymbol(f"neighbour_offset_{d}", create_type(np.int64)) for d in ['x', 'y', 'z'][:dim]]
def __init__(self, stencil, offsets_dtype=np.int32): def __init__(self, stencil, offsets_dtype=np.int64):
offsets_dtype = create_type(offsets_dtype) offsets_dtype = create_type(offsets_dtype)
dim = len(stencil[0]) dim = len(stencil[0])
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment