Skip to content

[rewriter] Decouple llama rule sets #2128

Closed
@titaiwangms

Description

@titaiwangms
Contributor

https://github.com/microsoft/onnxscript/blob/main/onnxscript/rewriter/llama_rule_sets.py

The file contains a lot of rewrite rules, but it's name is misleading, and its test (https://github.com/microsoft/onnxscript/blob/main/onnxscript/rewriter/llama_rule_sets_test.py) is also tangled. We need the API in optimizer explicitly specify what rules we are applying.

_DEFAULT_REWRITE_RULES = [

Activity

changed the title [-][rewriter] llama rule sets[/-] [+][rewriter] Decouple llama rule sets[/+] on Mar 24, 2025
added theissue type on Mar 24, 2025
added a commit that references this issue on Jun 18, 2025
f62f3bc
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

Projects

No projects

Milestone

No milestone

Relationships

None yet

    Development

    Participants

    @titaiwangms

    Issue actions

      [rewriter] Decouple llama rule sets · Issue #2128 · microsoft/onnxscript