pytorch - 💡(How to fix) Fix torch.export.export accepts a parameter whose storage is too small, then save/runtime fails with low-level setStorage error

import io
import platform
import sys
import traceback

import torch
import torch.nn as nn

print("python:", sys.version.split()[0])
print("platform:", platform.platform())
print("torch:", torch.__version__)
print("cuda_available:", torch.cuda.is_available())

x = torch.ones(1, 4)
m = nn.Linear(4, 4)
with torch.no_grad():
    m.weight.fill_(7.0)
    m.bias.zero_()

print("clean_eager_output:", m(x).tolist())
print("weight_shape:", tuple(m.weight.shape))
print("weight_numel:", m.weight.numel())
print("storage_nbytes_before:", m.weight.untyped_storage().nbytes())

m.weight.data.untyped_storage().resize_(0)
print("storage_nbytes_after:", m.weight.untyped_storage().nbytes())

try:
    m(x)
    print("corrupt_eager_status: unexpectedly succeeded")
except Exception as e:
    print("corrupt_eager_status: raised", type(e).__name__, str(e))

try:
    ep = torch.export.export(m, (x,))
    print("export_status: succeeded")
except Exception as e:
    print("export_status: raised", type(e).__name__, str(e))
    raise SystemExit

try:
    ep.module()(x)
    print("exported_module_status: unexpectedly succeeded")
except Exception as e:
    print("exported_module_status: raised", type(e).__name__, str(e))

try:
    buf = io.BytesIO()
    torch.export.save(ep, buf)
    print("save_status: unexpectedly succeeded")
except Exception as e:
    print("save_status: raised")
    print("exception_type:", type(e).__name__)
    print("exception_message:", str(e))
    traceback.print_exc(limit=10)

---

python: 3.13.5
platform: Linux-4.4.0-x86_64-with-glibc2.41
torch: 2.10.0+cpu
cuda_available: False
clean_eager_output: [[28.0, 28.0, 28.0, 28.0]]
weight_shape: (4, 4)
weight_numel: 16
storage_nbytes_before: 64
storage_nbytes_after: 0
corrupt_eager_status: raised RuntimeError setStorage: sizes [4, 4], strides [1, 4], storage offset 0, and itemsize 4 requiring a storage size of 64 are out of bounds for storage of size 0
export_status: succeeded
exported_module_status: raised RuntimeError setStorage: sizes [4, 4], strides [1, 4], storage offset 0, and itemsize 4 requiring a storage size of 64 are out of bounds for storage of size 0
save_status: raised
exception_type: RuntimeError
exception_message: setStorage: sizes [], strides [], storage offset 15, and itemsize 4 requiring a storage size of 64 are out of bounds for storage of size 0

---

File ".../torch/export/pt2_archive/_package_weights.py", line 11, in _end_ptr
    stop = tensor.view(-1)[-1].data_ptr() + tensor.element_size()
RuntimeError: setStorage: sizes [], strides [], storage offset 15, and itemsize 4 requiring a storage size of 64 are out of bounds for storage of size 0

---

python: 3.13.5
torch: 2.10.0+cpu
platform: Linux-4.4.0-x86_64-with-glibc2.41
cuda_available: False