Skip to content

[WIP] IdModel-based indexing#2238

Closed
naoyam wants to merge 484 commits intomainfrom
IterDomainGraphs_indexing2
Closed

[WIP] IdModel-based indexing#2238
naoyam wants to merge 484 commits intomainfrom
IterDomainGraphs_indexing2

Conversation

@naoyam
Copy link
Collaborator

@naoyam naoyam commented May 13, 2024

(Rebased #2218)

@naoyam naoyam added the idmodel label May 20, 2024
naoyam added a commit that referenced this pull request May 20, 2024
(Extracted from #2238)

Traversal for finding the shortest path from ValGroups to another
ValGroups. The algorithm is based on the standard BFS traversal,
however, since ValGraph is not an undirected graph, the dependencies of
ValGroups and ExprGroups need to be
satisfied. Specifically, when visiting an ExprGroup, either its inputs
or outputs must be visited before. Similarly, when visiting
a ValGroup, there must be at least one defining ExprGroup or one use
ExprGroup that is already visited.

The main use case is tensor indexing (#2238), where a typical traversal
would be from loop domains to allocation domains. Some indexing-specific
specialization would be needed, for example, dependencies with broadcast
domains can be ignored as their index is always just zero. The indexing
shortest-path traversal would be implemented by subclassing this class.

Please see the new unit tests for concrete examples.
@naoyam
Copy link
Collaborator Author

naoyam commented Apr 9, 2025

!test

@naoyam
Copy link
Collaborator Author

naoyam commented Apr 9, 2025

!test

1 similar comment
@naoyam
Copy link
Collaborator Author

naoyam commented Apr 9, 2025

!test

@naoyam
Copy link
Collaborator Author

naoyam commented Apr 10, 2025

!test

1 similar comment
@naoyam
Copy link
Collaborator Author

naoyam commented Apr 10, 2025

!test

@naoyam
Copy link
Collaborator Author

naoyam commented Apr 10, 2025

!test --diff

@naoyam
Copy link
Collaborator Author

naoyam commented Apr 11, 2025

!test

@naoyam
Copy link
Collaborator Author

naoyam commented Apr 12, 2025

!test

@naoyam
Copy link
Collaborator Author

naoyam commented Apr 14, 2025

!test

@naoyam
Copy link
Collaborator Author

naoyam commented Apr 14, 2025

!test

@naoyam
Copy link
Collaborator Author

naoyam commented Apr 15, 2025

!test

@naoyam naoyam closed this Feb 3, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant