Skip to content

feat: run-time index tensors and independent-axis tiling for reductions and data movement - #129

Merged
asteinh merged 2 commits into
developfrom
feat/tile-reductions
Oct 4, 2026
Merged

asteinh merged 2 commits into
developfrom
feat/tile-reductions

Conversation

@asteinh

@asteinh asteinh commented Oct 4, 2026

Copy link
Copy Markdown
Member
  • int32 index tensors may come from a model input or an ArgMax/ArgMin output and feed only index operands; data stays float32 or int8. ONNX int64 index inputs narrow to int32 at the boundary.
  • Gather-family and DynamicUpdateSlice metadata support run-time indices next to the constant form.
  • The partitioner tiles reductions, ArgMax/ArgMin and index-based data movement along an axis they leave independent, in height or matrix-row bands. Run-time-indexed operators stay untiled.
  • Contract cases: all four index operators with model-input indices, and ArgMax feeding Gather, in float32 and int8.

Needs the runtime branch of the same name.

@asteinh
asteinh merged commit d6bb64c into develop Oct 4, 2026
24 checks passed
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