Skip to content

Keep MLX dense event-rate state on device - #35

Merged
cboulay merged 1 commit into
devfrom
fix/mlx-dense-event-pipeline
Jul 22, 2026
Merged

Keep MLX dense event-rate state on device#35
cboulay merged 1 commit into
devfrom
fix/mlx-dense-event-pipeline

Conversation

@cboulay

@cboulay cboulay commented Jul 22, 2026

Copy link
Copy Markdown
Member

Summary

  • keep dense COUNT + SUM carry state in the input array namespace to avoid per-chunk MLX device synchronization
  • reset threshold state after an empty first chunk so the next MLX chunk has a valid previous-sample reference
  • preserve sparse threshold output by default and require explicit DENSE configuration for accelerator pipelines
  • add NumPy and MLX regression coverage for output contracts, empty-first-chunk handling, and end-to-end dense event-rate processing

Testing

  • uv run pytest -q (155 passed)
  • uv run ruff check src tests
  • git diff --check dev --

@cboulay
cboulay merged commit 41ed3d3 into dev Jul 22, 2026
4 checks passed
@cboulay
cboulay deleted the fix/mlx-dense-event-pipeline branch July 22, 2026 14:27
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant