Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

opt-125M-awq onnx model #258

Open
AmosLewis opened this issue Jun 12, 2024 · 1 comment
Open

opt-125M-awq onnx model #258

AmosLewis opened this issue Jun 12, 2024 · 1 comment
Assignees

Comments

@AmosLewis
Copy link
Collaborator

command.log

PYTHONPATH=/home/chi/src/torch-mlir/build/tools/torch-mlir/python_packages/torch_mlir python runmodel.py  --todtype default --mode onnx --outfileprefix opt-125M-awq 1> model-run.log 2>&1
PYTHONPATH=/home/chi/src/torch-mlir/build/tools/torch-mlir/python_packages/torch_mlir python -m torch_mlir.tools.import_onnx /home/chi/src/SHARK-TestSuite/e2eshark/onnx/models/opt-125M-awq/model.onnx -o opt-125M-awq.default.torch-onnx.mlir 1> onnx-import.log 2>&1
/home/chi/src/torch-mlir/build/bin/torch-mlir-opt -pass-pipeline='builtin.module(func.func(convert-torch-onnx-to-torch),torch-lower-to-backend-contract,func.func(cse,canonicalize),torch-backend-to-linalg-on-tensors-backend-pipeline)' opt-125M-awq.default.torch-onnx.mlir > opt-125M-awq.default.onnx.linalg.mlir 2>torch-mlir.log

torch-mlir.log

opt-125M-awq.default.torch-onnx.mlir:1123:13: error: 'tensor.expand_shape' op expected number of static shape dims to be equal to the output rank (3) but found 0 inputs instead
    %1119 = torch.operator "onnx.Gather"(%0, %1113) : (!torch.vtensor<[50272,768],f32>, !torch.vtensor<[?,?],si64>) -> !torch.vtensor<[?,?,768],f32> 
            ^
opt-125M-awq.default.torch-onnx.mlir:1123:13: note: see current operation: %1049 = "tensor.expand_shape"(%1046) <{reassociation = [[0, 1], [2]], static_output_shape = array<i64>}> : (tensor<?x768xf32>) -> tensor<?x?x768xf32>

Related issues: nod-ai/SHARK-ModelDev#242

@AmosLewis
Copy link
Collaborator Author

New iree issue after gather : iree-org/iree#17806

@AmosLewis AmosLewis assigned zjgarvey and unassigned Shukla-Gaurav Jul 3, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants