OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[ONNX] Support aten::bit_wise_not in fx-onnx exporter (#94919)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/94919
Approved by: https://github.com/justinchuby, https://github.com/wschin
This commit is contained in:
AllenTiTaiWang 2023-02-15 23:06:27 +00:00 committed by PyTorch MergeBot
parent a0389681c2
commit 28e69954a1
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.ci/onnx/test.sh
View File

@ -64,7 +64,7 @@ if [[ "$BUILD_ENVIRONMENT" == *onnx* ]]; then
# TODO: change this when onnx 1.13.1 is released.
pip install --no-use-pep517 'onnx @ git+https://github.com/onnx/onnx@e192ba01e438d22ca2dedd7956e28e3551626c91'
# TODO: change this when onnx-script is on testPypi
pip install 'onnx-script @ git+https://github.com/microsoft/onnx-script@a71e35bcd72537bf7572536ee57250a0c0488bf6'
pip install 'onnx-script @ git+https://github.com/microsoft/onnx-script@0298154caf6b46fc4e30abba034095c1290c26e3'
# numba requires numpy <= 1.20, onnxruntime requires numpy >= 1.21.
# We don't actually need it for our tests, but it's imported if it's present, so uninstall.
pip uninstall -q --yes numba

2
torch/onnx/_internal/fx/function_dispatcher.py
View File

@ -49,7 +49,7 @@ _ATENLIB_FUNCTIONS = {
"aten::atan": ops.core.aten_atan,
"aten::atanh": ops.core.aten_atanh,
"aten::baddbmm": ops.core.aten_baddbmm,
"aten::bitwise_not": ops.core.aten_bitwise_not,
"aten::bitwise_not": ops.core.aten_bitwise_not_bool,
"aten::bmm": ops.core.aten_bmm,
"aten::ceil": ops.core.aten_ceil,
"aten::celu": ops.nn.aten_celu,