Skip to content

Expose OpBuilder as onnxscript.OpBuilder#2840

Merged
justinchuby merged 1 commit intomainfrom
rama/opbuilder
Mar 4, 2026
Merged

Expose OpBuilder as onnxscript.OpBuilder#2840
justinchuby merged 1 commit intomainfrom
rama/opbuilder

Conversation

@gramalingam
Copy link
Collaborator

Add OpBuilder to all and import it from _internal.builder in the top-level onnxscript package.

Add OpBuilder to __all__ and import it from _internal.builder
in the top-level onnxscript package.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
@codecov
Copy link

codecov bot commented Mar 4, 2026

❌ 1 Tests Failed:

Tests completed Failed Passed Skipped
13061 1 13060 2257
View the top 1 failed test(s) by shortest run time
onnxscript.rewriter.ort_fusions.rms_normalization_test.TestRmsNormalization::test_smollm
Stack Traces | 16.5s run time
onnxscript\rewriter\ort_fusions\rms_normalization_test.py:23: in test_smollm
    new_outputs = ort_run("optimized", model, inputs)
                  ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
onnxscript\rewriter\ort_fusions\_test_utils.py:19: in ort_run
    model_proto.SerializeToString(), options, providers=providers
    ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
E   google.protobuf.message.EncodeError: Failed to serialize proto
View the full list of 1 ❄️ flaky test(s)
tests.function_libs.torch_lib.ops_test.TestOutputConsistencyFullGraphCPU::test_output_match_opinfo__logsumexp_cpu_float16

Flake rate in main: 14.58% (Passed 252 times, Failed 43 times)

Stack Traces | 0.165s run time
.../function_libs/torch_lib/ops_test.py:243: in run_test_output_match
    torch.testing.assert_close(
E   AssertionError: Tensor-likes are not close!
E   
E   Mismatched elements: 1 / 5 (20.0%)
E   Greatest absolute difference: 2.288818359375e-05 at index (1,) (up to 1e-05 allowed)
E   Greatest relative difference: 0.0022869110107421875 at index (1,) (up to 0.001 allowed)

To view more test analytics, go to the Test Analytics Dashboard
📋 Got 3 mins? Take this short survey to help us improve Test Analytics.

@github-project-automation github-project-automation bot moved this from Todo to Done in ONNX Script Review Board Mar 4, 2026
@justinchuby justinchuby merged commit 6bfbb4d into main Mar 4, 2026
26 of 31 checks passed
@justinchuby justinchuby deleted the rama/opbuilder branch March 4, 2026 19:38
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

Development

Successfully merging this pull request may close these issues.

2 participants