Skip to content

[rewriter] Decouple llama rule sets #2128

Open
@titaiwangms

Description

@titaiwangms

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 = [

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions