From 2d46dfbc9a028e3731b30facd35d9d836a0ce9b5 Mon Sep 17 00:00:00 2001 From: chen-tianyu19 Date: Thu, 24 Aug 2023 13:11:13 +0800 Subject: [PATCH] fix infer shape for empty tensor in convolution --- torch_npu/csrc/framework/utils/KernelNpuOutputSize.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/torch_npu/csrc/framework/utils/KernelNpuOutputSize.cpp b/torch_npu/csrc/framework/utils/KernelNpuOutputSize.cpp index 1de3c52de2..fe955737d3 100644 --- a/torch_npu/csrc/framework/utils/KernelNpuOutputSize.cpp +++ b/torch_npu/csrc/framework/utils/KernelNpuOutputSize.cpp @@ -347,6 +347,7 @@ namespace native { int64_t N = input.size(0); int64_t L = input.size(2); int64_t C_out = weight.size(0); + C_out = (weight.size(1) != 0) ? C_out : 0; auto kernel_size = weight.sizes().slice(2); -- Gitee