Skip to content

Add GEN_INDEX pointwise op#359

Open
rsuderman wants to merge 2 commits intoiree-org:mainfrom
rsuderman:pointwise_gen_idx
Open

Add GEN_INDEX pointwise op#359
rsuderman wants to merge 2 commits intoiree-org:mainfrom
rsuderman:pointwise_gen_idx

Conversation

@rsuderman
Copy link
Copy Markdown
Contributor

Emits the output as a linalg.generic that materializes the iteration index along a configurable axis (via PointwiseAttr::setGenIdxAxis). The generic is bridged back to torch via torch_c.from_builtin_tensor so the result feeds the standard output permute.

Also adds buildBuiltinTensorTypeStr / getSignlessElementTypeAsm / getIdentityAffineMapAsm / getIteratorTypesAsm helpers and relaxes the lit-test MLIR indentation check to allow nested region bodies.

Emits the output as a linalg.generic that materializes the iteration index
along a configurable axis (via PointwiseAttr::setGenIdxAxis). The generic
is bridged back to torch via torch_c.from_builtin_tensor so the result
feeds the standard output permute.

Also adds buildBuiltinTensorTypeStr / getSignlessElementTypeAsm /
getIdentityAffineMapAsm / getIteratorTypesAsm helpers and relaxes the
lit-test MLIR indentation check to allow nested region bodies.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
Signed-off-by: Rob Suderman <rob.suderman@gmail.com>
Comment thread include/fusilli/support/asm_emitter.h Outdated
@sjain-stanford sjain-stanford requested a review from IanWood1 April 20, 2026 20:30
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.

2 participants