diff --git a/src/mindtorch/_apis/npu_910a.py b/src/mindtorch/_apis/npu_910a.py index 4b4e002a8..965e12a77 100644 --- a/src/mindtorch/_apis/npu_910a.py +++ b/src/mindtorch/_apis/npu_910a.py @@ -1536,6 +1536,8 @@ def cdist(x1, x2, p): return legacy.cdist(x1, x2, float(p)) def unstack_view(input, dim): + if input.shape[dim] == 0: + return tuple() if ENABLE_PYBOOST: return pyboost.unstack_ext_view_op(input, dim) return legacy.unstack(input, dim, input.shape[dim]) diff --git a/src/mindtorch/_apis/npu_910b.py b/src/mindtorch/_apis/npu_910b.py index bcb158a50..cd1d00754 100644 --- a/src/mindtorch/_apis/npu_910b.py +++ b/src/mindtorch/_apis/npu_910b.py @@ -1526,6 +1526,8 @@ def cdist(x1, x2, p): return legacy.cdist(x1, x2, float(p)) def unstack_view(input, dim): + if input.shape[dim] == 0: + return tuple() if ENABLE_PYBOOST: return pyboost.unstack_ext_view_op(input, dim) return legacy.unstack(input, dim, input.shape[dim])