Skip to content

[rewriter] Decouple llama rule sets #2128

Closed
@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

Labels

No labels
No labels

Type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions