You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
iree candidate-20240610.920
torch-mlir
commit 7e0e23c66820d1db548103acbdf1337f701dc5a3 (upstream/main)
Author: Sambhav Jain <[email protected]>
Date: Sun Jun 9 00:32:49 2024 -0700
Test custom op import with symbolic shapes (#3431)
Tests the basic constructs of registering a custom op and its abstract
implementations (with FakeTensors) in python, going through TorchDynamo
export, followed by importing the shape expressions in the Torch
dialect.
Also fixes the importer were previously the symbolic bind op insertion
was not gated in one place.
model-run.log
Traceback (most recent call last):
File "/home/chi/src/SHARK-TestSuite/e2eshark/test-run/pytorch/models/opt-125m-gptq/runmodel.py", line 149, in <module>
onnx_program = torch.onnx.export(model, E2ESHARK_CHECK["input"], onnx_name)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/chi/src/torch-mlir/mlir_venv/lib/python3.11/site-packages/torch/onnx/utils.py", line 516, in export
_export(
File "/home/chi/src/torch-mlir/mlir_venv/lib/python3.11/site-packages/torch/onnx/utils.py", line 1612, in _export
graph, params_dict, torch_out = _model_to_graph(
^^^^^^^^^^^^^^^^
File "/home/chi/src/torch-mlir/mlir_venv/lib/python3.11/site-packages/torch/onnx/utils.py", line 1138, in _model_to_graph
graph = _optimize_graph(
^^^^^^^^^^^^^^^^
File "/home/chi/src/torch-mlir/mlir_venv/lib/python3.11/site-packages/torch/onnx/utils.py", line 677, in _optimize_graph
graph = _C._jit_pass_onnx(graph, operator_export_type)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/chi/src/torch-mlir/mlir_venv/lib/python3.11/site-packages/torch/onnx/utils.py", line 1969, in _run_symbolic_function
raise errors.UnsupportedOperatorError(
torch.onnx.errors.UnsupportedOperatorError: Exporting the operator 'aten::bitwise_right_shift' to ONNX opset version 17 is not supported. Please feel free to request support or submit a pull request on PyTorch GitHub: https://github.com/pytorch/pytorch/issues.
The text was updated successfully, but these errors were encountered:
python ./run.py --torchmlirbuild ../../torch-mlir/build --tolerance 0.001 0.001 --cachedir ./huggingface_cache --ireebuild ../../iree-build -f pytorch -g models --mode onnx --report --torchtolinalg --tests pytorch/models/pytorch/models/opt-125m-gptq
model-run.log
The text was updated successfully, but these errors were encountered: