Open
Description
🐛 Describe the bug
Bug Code 1:
import torch
from torch.nn.utils.rnn import pad_packed_sequence, PackedSequence
empty_data = torch.randn(0, 5)
empty_batch_sizes = torch.tensor([], dtype=torch.int64)
empty_packed = PackedSequence(empty_data, empty_batch_sizes, None, None)
pad_packed_sequence(empty_packed, batch_first=True)
Output for Bug Code 1:
Segmentation fault
Bug Code 2:
import torch
from torch.nn.utils.rnn import PackedSequence, unpack_sequence
empty_data = torch.tensor([])
empty_batch_sizes = torch.tensor([], dtype=torch.int64)
packed = PackedSequence(data=empty_data, batch_sizes=empty_batch_sizes)
unpack_sequence(packed)
Output for Bug Code 2:
Segmentation fault
Versions
torch 2.6.0
cc @malfet