Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add rotary embedding fusion rule (part 1) #1981

Merged
merged 23 commits into from
Jan 7, 2025

Minor fixes

a745039
Select commit
Loading
Failed to load commit list.
Merged

Add rotary embedding fusion rule (part 1) #1981

Minor fixes
a745039
Select commit
Loading
Failed to load commit list.
Codecov / codecov/patch succeeded Dec 21, 2024 in 1s

81.17% of diff hit (target 74.73%)

View this Pull Request on Codecov

81.17% of diff hit (target 74.73%)

Annotations

Check warning on line 314 in onnxscript/optimizer/_constant_folding.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/optimizer/_constant_folding.py#L314 

Added line #L314 was not covered by tests

Check warning on line 317 in onnxscript/optimizer/_constant_folding.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/optimizer/_constant_folding.py#L317 

Added line #L317 was not covered by tests

Check warning on line 320 in onnxscript/optimizer/_constant_folding.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/optimizer/_constant_folding.py#L320 

Added line #L320 was not covered by tests

Check warning on line 323 in onnxscript/optimizer/_constant_folding.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/optimizer/_constant_folding.py#L323 

Added line #L323 was not covered by tests

Check warning on line 91 in onnxscript/rewriter/_ir_utils.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/_ir_utils.py#L91 

Added line #L91 was not covered by tests

Check warning on line 98 in onnxscript/rewriter/_ir_utils.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/_ir_utils.py#L97-L98 

Added lines #L97 - L98 were not covered by tests

Check warning on line 104 in onnxscript/rewriter/_ir_utils.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/_ir_utils.py#L104 

Added line #L104 was not covered by tests

Check warning on line 50 in onnxscript/rewriter/onnxruntime/xformers/cos_sin_cache.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/onnxruntime/xformers/cos_sin_cache.py#L50 

Added line #L50 was not covered by tests

Check warning on line 52 in onnxscript/rewriter/onnxruntime/xformers/cos_sin_cache.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/onnxruntime/xformers/cos_sin_cache.py#L52 

Added line #L52 was not covered by tests

Check warning on line 55 in onnxscript/rewriter/onnxruntime/xformers/cos_sin_cache.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/onnxruntime/xformers/cos_sin_cache.py#L55 

Added line #L55 was not covered by tests

Check warning on line 29 in onnxscript/rewriter/onnxruntime/xformers/cos_sin_cache_test.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/onnxruntime/xformers/cos_sin_cache_test.py#L29 

Added line #L29 was not covered by tests

Check warning on line 33 in onnxscript/rewriter/onnxruntime/xformers/rotary_embedding.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/onnxruntime/xformers/rotary_embedding.py#L33 

Added line #L33 was not covered by tests

Check warning on line 35 in onnxscript/rewriter/onnxruntime/xformers/rotary_embedding.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/onnxruntime/xformers/rotary_embedding.py#L35 

Added line #L35 was not covered by tests

Check warning on line 38 in onnxscript/rewriter/onnxruntime/xformers/rotary_embedding.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/onnxruntime/xformers/rotary_embedding.py#L38 

Added line #L38 was not covered by tests

Check warning on line 23 in onnxscript/rewriter/onnxruntime/xformers/rotary_embedding_test.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/onnxruntime/xformers/rotary_embedding_test.py#L23 

Added line #L23 was not covered by tests

Check warning on line 1469 in onnxscript/rewriter/pattern.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/pattern.py#L1469 

Added line #L1469 was not covered by tests
View more details on Codecov