-
Notifications
You must be signed in to change notification settings - Fork 68
[Bug] ops.concat does not work the same as torch.cat #440
Copy link
Copy link
Closed
Labels
TaskOr "Story" in JIRA's term.Or "Story" in JIRA's term.bugSomething isn't workingSomething isn't working
Milestone
Description
Describe the bug
ops.concat does not work the same as torch.cat on an empty tensor(shape=[0]), and this cause an Runtime error when interpreting torch.cat.
To Reproduce
The following is my code for test:
import torch
import hidet
from hidet.graph import ops
x = torch.rand([3, 4, 5])
y = torch.rand([0])
print(torch.cat([x, y]))
# x = hidet.randn([3, 4, 5])
# y = hidet.randn([0])
# print(ops.concat([x, y], axis = 0))The hidet code report an error but torch work normally.
Expected behavior
I hope that ops.cat can do as tensor.cat does.
Or how can I custom what interpreter dose, for example, interpret torch.cat as my custome operator, but not ops.concat defaultly.
Enviroment
- OS: Ubuntu 22.04
- hidet version: 0.3.0
- torch version: 2.2.1 cuda
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
TaskOr "Story" in JIRA's term.Or "Story" in JIRA's term.bugSomething isn't workingSomething isn't working