From f162523c2895c458c28409137c147dbbc71d1ead Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E6=99=93?= Date: Wed, 13 Apr 2022 09:42:40 +0000 Subject: [PATCH] fix atan2 --- torch_npu/csrc/aten/ops/Atan2KernelNpu.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch_npu/csrc/aten/ops/Atan2KernelNpu.cpp b/torch_npu/csrc/aten/ops/Atan2KernelNpu.cpp index 433cac4f6a1..3a8b976fb45 100644 --- a/torch_npu/csrc/aten/ops/Atan2KernelNpu.cpp +++ b/torch_npu/csrc/aten/ops/Atan2KernelNpu.cpp @@ -45,7 +45,7 @@ at::Tensor& NPUNativeFunctions::atan2_out( outputSize); if (!NpuUtils::check_match(&result)) { at::Tensor contiguousResult = NpuUtils::format_contiguous(result); - at::Tensor result = atan2_out_npu_nocheck(self, other, contiguousResult); + atan2_out_npu_nocheck(self, other, contiguousResult); NpuUtils::format_fresh_view(result, contiguousResult); } else { atan2_out_npu_nocheck(self, other, result); -- Gitee