Skip to content

Trace single op.SplitToSequence#2817

Merged
titaiwangms merged 1 commit into
microsoft:mainfrom
titaiwangms:titaiwang/trace_splitto_seq
Feb 19, 2026
Merged

Trace single op.SplitToSequence#2817
titaiwangms merged 1 commit into
microsoft:mainfrom
titaiwangms:titaiwang/trace_splitto_seq

Conversation

@titaiwangms
Copy link
Copy Markdown
Contributor

This pull request updates several operator registrations in onnxscript/function_libs/torch_lib/ops/core.py to mark them as trace_only, which affects how these ops are handled during tracing and export. The most important changes are:

Operator registration updates (trace-only):

  • Marked the aten::split and aten::split.Tensor operators as trace_only in the aten_split function registration.
  • Marked the aten::split_with_sizes operator as trace_only in the aten_split_with_sizes function registration.
  • Marked the aten::unsafe_split.Tensor operator as trace_only in the aten_unsafe_split function registration.

@titaiwangms titaiwangms added the module: torchlib Related to the torch/aten function lib in development label Feb 19, 2026
@titaiwangms titaiwangms enabled auto-merge (squash) February 19, 2026 17:55
@gramalingam
Copy link
Copy Markdown
Collaborator

I am curious what motivated this. What does this change? Is there a unit test, if this is fixing a bug?

@codecov
Copy link
Copy Markdown

codecov Bot commented Feb 19, 2026

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 70.42%. Comparing base (0d06d3b) to head (60f296b).
⚠️ Report is 1 commits behind head on main.
✅ All tests successful. No failed tests found.

Additional details and impacted files
@@            Coverage Diff             @@
##             main    #2817      +/-   ##
==========================================
+ Coverage   70.40%   70.42%   +0.01%     
==========================================
  Files         228      228              
  Lines       27140    27140              
  Branches     2727     2727              
==========================================
+ Hits        19108    19113       +5     
+ Misses       7096     7091       -5     
  Partials      936      936              

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

@titaiwangms titaiwangms disabled auto-merge February 19, 2026 18:24
@titaiwangms titaiwangms merged commit 8c176f2 into microsoft:main Feb 19, 2026
31 of 32 checks passed
@titaiwangms titaiwangms deleted the titaiwang/trace_splitto_seq branch February 19, 2026 19:11
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.

4 participants