Skip to content
This repository was archived by the owner on Jan 31, 2025. It is now read-only.

fix halo exchange #30

Merged
merged 2 commits into from
Apr 29, 2024
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 10 additions & 9 deletions pyop3/lang.py
Original file line number Diff line number Diff line change
Expand Up @@ -360,6 +360,16 @@ def _buffer_exchanges(buffer, intent, *, touches_ghost_points):
assert buffer._pending_reduction is not None
initializers.append(buffer._reduce_leaves_to_roots_begin)
reductions.append(buffer._reduce_leaves_to_roots_end)
# set leaves to appropriate nil value
if intent == INC:
def _zero():
buffer._data[buffer.sf.ileaf] = 0
reductions.append(_zero)
elif intent in {MIN_WRITE, MIN_RW}:
buffer._data[buffer.sf.ileaf] = dtype_limits(buffer.dtype).max
else:
assert intent in {MAX_WRITE, MAX_RW}
buffer._data[buffer.sf.ileaf] = dtype_limits(buffer.dtype).min

# We are modifying owned values so the leaves must now be wrong
buffer._leaves_valid = False
Expand All @@ -370,15 +380,6 @@ def _buffer_exchanges(buffer, intent, *, touches_ghost_points):
else:
buffer._pending_reduction = intent

# set leaves to appropriate nil value
if intent == INC:
buffer._data[buffer.sf.ileaf] = 0
elif intent in {MIN_WRITE, MIN_RW}:
buffer._data[buffer.sf.ileaf] = dtype_limits(buffer.dtype).max
else:
assert intent in {MAX_WRITE, MAX_RW}
buffer._data[buffer.sf.ileaf] = dtype_limits(buffer.dtype).min

return tuple(initializers), tuple(reductions), tuple(broadcasts)


Expand Down
Loading