Skip to content

Fix include_self for scatter_reduce#2090

Merged
justinchuby merged 36 commits intomicrosoft:mainfrom
xadupre:scatter
Mar 31, 2025
Merged

Fix include_self for scatter_reduce#2090
justinchuby merged 36 commits intomicrosoft:mainfrom
xadupre:scatter

Conversation

@xadupre
Copy link
Copy Markdown
Member

@xadupre xadupre commented Mar 7, 2025

Implement logic for include_self. Fixes pytorch/pytorch#147617

@codecov
Copy link
Copy Markdown

codecov Bot commented Mar 7, 2025

Codecov Report

Attention: Patch coverage is 87.87879% with 4 lines in your changes missing coverage. Please review.

Project coverage is 72.99%. Comparing base (5d969c4) to head (dfd613b).
Report is 2 commits behind head on main.

Files with missing lines Patch % Lines
onnxscript/function_libs/torch_lib/ops/core.py 86.20% 2 Missing and 2 partials ⚠️
Additional details and impacted files
@@            Coverage Diff             @@
##             main    #2090      +/-   ##
==========================================
+ Coverage   72.93%   72.99%   +0.05%     
==========================================
  Files         223      223              
  Lines       29484    29514      +30     
  Branches     3483     3492       +9     
==========================================
+ Hits        21504    21543      +39     
+ Misses       6813     6802      -11     
- Partials     1167     1169       +2     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

@justinchuby justinchuby self-assigned this Mar 7, 2025
@justinchuby justinchuby self-requested a review March 7, 2025 16:54
Comment thread onnxscript/function_libs/torch_lib/ops/core.py Outdated
Comment thread onnxscript/function_libs/torch_lib/ops/core.py Outdated
Comment thread onnxscript/function_libs/torch_lib/ops/core.py
Comment thread onnxscript/function_libs/torch_lib/ops/core.py
Comment thread onnxscript/function_libs/torch_lib/ops/core.py Outdated
Comment thread onnxscript/function_libs/torch_lib/ops/core.py Outdated
Comment thread tests/function_libs/torch_lib/ops_test_data.py Outdated
Comment thread onnxscript/function_libs/torch_lib/ops/core.py Fixed
Comment thread onnxscript/function_libs/torch_lib/ops/core.py Fixed
Comment thread onnxscript/function_libs/torch_lib/ops/core.py Outdated
Comment thread onnxscript/function_libs/torch_lib/ops/core.py Fixed
Comment thread onnxscript/function_libs/torch_lib/ops/core.py Fixed
@justinchuby justinchuby added the module: torchlib Related to the torch/aten function lib in development label Mar 14, 2025
Comment thread tests/function_libs/torch_lib/e2e_ops_tests.py Fixed
Comment thread tests/function_libs/torch_lib/e2e_ops_tests.py Fixed
Comment thread tests/function_libs/torch_lib/e2e_ops_tests.py Fixed
Comment thread tests/function_libs/torch_lib/e2e_ops_tests.py Fixed
Co-authored-by: Justin Chu <justinchuby@users.noreply.github.com>
Comment thread tests/function_libs/torch_lib/e2e_ops_tests.py Fixed
Comment thread tests/function_libs/torch_lib/e2e_ops_tests.py Fixed
@justinchuby
Copy link
Copy Markdown
Collaborator

I will take a look at the tests

Comment thread onnxscript/ir/_enums.py Outdated
Comment thread onnxscript/optimizer/_constant_folding.py Outdated
Comment thread onnxscript/function_libs/torch_lib/ops/core.py Fixed
Comment thread onnxscript/function_libs/torch_lib/ops/core.py Fixed
Comment thread onnxscript/function_libs/torch_lib/ops/core.py Outdated
Comment thread onnxscript/function_libs/torch_lib/ops/core.py Outdated
Comment thread onnxscript/function_libs/torch_lib/ops/core.py Outdated
@justinchuby justinchuby enabled auto-merge (squash) March 31, 2025 19:44
@justinchuby justinchuby disabled auto-merge March 31, 2025 19:56
@justinchuby justinchuby enabled auto-merge (squash) March 31, 2025 19:57
@justinchuby justinchuby disabled auto-merge March 31, 2025 19:57
@justinchuby justinchuby enabled auto-merge (squash) March 31, 2025 19:59
@justinchuby justinchuby merged commit 9ee8c92 into microsoft:main Mar 31, 2025
24 of 29 checks passed
bmehta001 pushed a commit to bmehta001/onnxscript that referenced this pull request Apr 11, 2025
Implement logic for include_self. Fixes
pytorch/pytorch#147617

---------

Co-authored-by: Justin Chu <justinchuby@users.noreply.github.com>
Co-authored-by: Ti-Tai Wang <titaiwang@microsoft.com>
bmehta001 pushed a commit to bmehta001/onnxscript that referenced this pull request Apr 11, 2025
Implement logic for include_self. Fixes
pytorch/pytorch#147617

---------

Co-authored-by: Justin Chu <justinchuby@users.noreply.github.com>
Co-authored-by: Ti-Tai Wang <titaiwang@microsoft.com>
bmehta001 pushed a commit to bmehta001/onnxscript that referenced this pull request Apr 11, 2025
Implement logic for include_self. Fixes
pytorch/pytorch#147617

---------

Co-authored-by: Justin Chu <justinchuby@users.noreply.github.com>
Co-authored-by: Ti-Tai Wang <titaiwang@microsoft.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

module: torchlib Related to the torch/aten function lib in development

Projects

Development

Successfully merging this pull request may close these issues.

[ONNX] GNN model inaccuracy: scatter_reduce need to be fixed

4 participants