masahi commented on issue #10651:
URL: https://github.com/apache/tvm/issues/10651#issuecomment-1070494750
Yeah with this code I get the same error as onnx:
```
import torch
class Net(torch.nn.Module):
def __init__(self) -> None:
super().__init__()
self.weight = torch.nn.Parameter(torch.randn(3, 1))
def forward(self, x):
return torch.matmul(x, self.weight)
net = Net().eval()
import tvm
from tvm import relay
from tvm.relay.frontend import from_pytorch
scripted_model = torch.jit.trace(net, torch.randn((3,))).eval()
mod, params = from_pytorch(scripted_model, [("x", (3,))])
with tvm.transform.PassContext(opt_level=4):
relay.build_module.create_executor("graph", mod, tvm.cpu(),
target='llvm').evaluate()
```
```
File "/home/masa/projects/dev/tvm/python/tvm/topi/x86/dense_alter_op.py",
line 48, in _alter_dense_layout
M, K = get_const_tuple(data_tensor.shape)
ValueError: not enough values to unpack (expected 2, got 1)
```
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
