Skip to content

Refactor: use backend *_expr methods in PointerIndexingStrategy#1850

Merged
aditvenk merged 1 commit intomainfrom
aditvenk/stack/16
Mar 28, 2026
Merged

Refactor: use backend *_expr methods in PointerIndexingStrategy#1850
aditvenk merged 1 commit intomainfrom
aditvenk/stack/16

Conversation

@aditvenk
Copy link
Copy Markdown
Contributor

@aditvenk aditvenk commented Mar 27, 2026

Stacked PRs:


Refactor: use backend *_expr methods in PointerIndexingStrategy

Replace hardcoded tl.zeros, tl.broadcast_to, and tl.reshape calls in
PointerIndexingStrategy and SubscriptIndexing with backend.zeros_expr(),
backend.broadcast_to_expr(), and backend.reshape_expr() calls.
TritonBackend returns the same tl.* strings, so Triton codegen is
unchanged. This makes the shared indexing infrastructure backend-agnostic.

Replace hardcoded tl.zeros, tl.broadcast_to, and tl.reshape calls in
PointerIndexingStrategy and SubscriptIndexing with backend.zeros_expr(),
backend.broadcast_to_expr(), and backend.reshape_expr() calls.
TritonBackend returns the same tl.* strings, so Triton codegen is
unchanged. This makes the shared indexing infrastructure backend-agnostic.

stack-info: PR: #1850, branch: aditvenk/stack/16
@aditvenk aditvenk marked this pull request as draft March 28, 2026 00:10
@aditvenk aditvenk force-pushed the aditvenk/stack/16 branch from b8a13f6 to 081645f Compare March 28, 2026 00:12
@aditvenk aditvenk marked this pull request as ready for review March 28, 2026 00:14
@aditvenk aditvenk merged commit afa7a5c into main Mar 28, 2026
20 of 21 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Meta Open Source bot.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants