diff --git a/torch_npu/csrc/aten/common/TensorFactories.cpp b/torch_npu/csrc/aten/common/TensorFactories.cpp index 99711347ff0fa79680eccce78211c3be3e39f7e1..7b2c143313d9ac1708a1ed1e42af3077d26eacb5 100644 --- a/torch_npu/csrc/aten/common/TensorFactories.cpp +++ b/torch_npu/csrc/aten/common/TensorFactories.cpp @@ -776,6 +776,7 @@ namespace at_npu } at::Tensor& NPUNativeFunctions::full_out(at::IntArrayRef size, const at::Scalar& fill_value, at::Tensor& out) { + check_size_nonnegative(size); OpPreparation::CheckOut( {}, out,