Skip to content

SDPA fusion cleanup #2352

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

Merged
gramalingam merged 23 commits into from
Jun 3, 2025

Merge branch 'main' into rama/sdpa

06d3e9a
Select commit
Loading
Failed to load commit list.
Merged

SDPA fusion cleanup #2352

Merge branch 'main' into rama/sdpa
06d3e9a
Select commit
Loading
Failed to load commit list.
Codecov / codecov/patch succeeded Jun 3, 2025 in 1s

77.35% of diff hit (target 70.16%)

View this Pull Request on Codecov

77.35% of diff hit (target 70.16%)

Annotations

Check warning on line 34 in onnxscript/rewriter/_basics.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/_basics.py#L34 

Added line #L34 was not covered by tests

Check warning on line 30 in onnxscript/rewriter/_fusion_utils.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/_fusion_utils.py#L30 

Added line #L30 was not covered by tests

Check warning on line 32 in onnxscript/rewriter/_fusion_utils.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/_fusion_utils.py#L32 

Added line #L32 was not covered by tests

Check warning on line 104 in onnxscript/rewriter/ort_fusions/sdpa.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/ort_fusions/sdpa.py#L104 

Added line #L104 was not covered by tests

Check warning on line 121 in onnxscript/rewriter/ort_fusions/sdpa.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/ort_fusions/sdpa.py#L121 

Added line #L121 was not covered by tests

Check warning on line 86 in onnxscript/rewriter/ort_fusions/sdpa_test.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/ort_fusions/sdpa_test.py#L86 

Added line #L86 was not covered by tests

Check warning on line 98 in onnxscript/rewriter/ort_fusions/sdpa_test.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/ort_fusions/sdpa_test.py#L98 

Added line #L98 was not covered by tests

Check warning on line 111 in onnxscript/rewriter/ort_fusions/sdpa_test.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/ort_fusions/sdpa_test.py#L110-L111 

Added lines #L110 - L111 were not covered by tests

Check warning on line 123 in onnxscript/rewriter/ort_fusions/sdpa_test.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/ort_fusions/sdpa_test.py#L123 

Added line #L123 was not covered by tests

Check warning on line 134 in onnxscript/rewriter/ort_fusions/sdpa_test.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/ort_fusions/sdpa_test.py#L134 

Added line #L134 was not covered by tests

Check warning on line 195 in onnxscript/rewriter/ort_fusions/sdpa_test.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/ort_fusions/sdpa_test.py#L195 

Added line #L195 was not covered by tests

Check warning on line 208 in onnxscript/rewriter/ort_fusions/sdpa_test.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/ort_fusions/sdpa_test.py#L208 

Added line #L208 was not covered by tests

Check warning on line 221 in onnxscript/rewriter/ort_fusions/sdpa_test.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/ort_fusions/sdpa_test.py#L221 

Added line #L221 was not covered by tests

Check warning on line 233 in onnxscript/rewriter/ort_fusions/sdpa_test.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/ort_fusions/sdpa_test.py#L233 

Added line #L233 was not covered by tests

Check warning on line 286 in onnxscript/rewriter/ort_fusions/sdpa_test.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/ort_fusions/sdpa_test.py#L286 

Added line #L286 was not covered by tests

Check warning on line 293 in onnxscript/rewriter/ort_fusions/sdpa_test.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

onnxscript/rewriter/ort_fusions/sdpa_test.py#L292-L293 

Added lines #L292 - L293 were not covered by tests
View more details on Codecov