From 893a4bdd8a44144585d7d2ca10c98d1763fc2818 Mon Sep 17 00:00:00 2001 From: chen-tianyu19 Date: Thu, 7 Sep 2023 11:19:15 +0800 Subject: [PATCH] Fix full negative size --- torch_npu/csrc/aten/common/TensorFactories.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/torch_npu/csrc/aten/common/TensorFactories.cpp b/torch_npu/csrc/aten/common/TensorFactories.cpp index 99711347ff0..7b2c143313d 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, -- Gitee