[ONNX] Support aten::bit_wise_not in fx-onnx exporter

[titaiwangms]

Stack from ghstack (oldest at bottom):

• [ONNX] Refactor validation op-level #94920
• -> [ONNX] Support aten::bit_wise_not in fx-onnx exporter #94919
• [ONNX] Add bloom ops #94878

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/94919

• 📄 Preview Python docs built from this PR
• 📄 Preview C++ docs built from this PR
• ❓ Need help or want to give feedback on the CI? Visit the bot commands wiki or our office hours

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit ea118eb:
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[BowenBao]

Could you add a TODO/FIXME comment and/or tracking issue if this is temp?

[titaiwangms]

Could you add a TODO/FIXME comment and/or tracking issue if this is temp?

Haven't figured out a way on this ...
https://github.com/microsoft/onnx-script/pull/439/files#r1107520708

We don't have it in torch.onnx.export API actually... So maybe not a temp...

@_onnx_symbolic("aten::bitwise_not")
@_beartype.beartype
def bitwise_not(g: jit_utils.GraphContext, input):
    if not symbolic_helper._is_bool(input):
        raise errors.SymbolicValueError(
            "ONNX export does NOT support exporting bitwise Not "
            "for non-boolean input values",
            input,
        )
    return g.op("Not", input)

Concusion: ONNX export does NOT support exporting bitwise with INT dtype right now.

[titaiwangms]

@pytorchbot merge -g

[pytorchmergebot]

Merge started

Your change will be merged once all checks pass (ETA 0-4 Hours).

Learn more about merging in the wiki.

Questions? Feedback? Please reach out to the PyTorch DevX Team

Advanced Debugging

Check the merge workflow status
here