From bdbda358bc3adc9c4b26aa1ec4003f9eb5d1685e Mon Sep 17 00:00:00 2001 From: Feng Yuan Date: Mon, 28 Oct 2024 12:33:49 +0800 Subject: [PATCH] Windows: Fixing building error (#1019) Signed-off-by: Feng Yuan --- src/ATen/native/xpu/sycl/MaxUnpoolingKernels.h | 4 ++-- src/ATen/native/xpu/sycl/TensorFactoriesKernels.cpp | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/ATen/native/xpu/sycl/MaxUnpoolingKernels.h b/src/ATen/native/xpu/sycl/MaxUnpoolingKernels.h index 3fbd2b84e..8b6acfa3a 100644 --- a/src/ATen/native/xpu/sycl/MaxUnpoolingKernels.h +++ b/src/ATen/native/xpu/sycl/MaxUnpoolingKernels.h @@ -3,13 +3,13 @@ namespace at::native::xpu { -Tensor& max_unpooling2d_forward_kernel( +TORCH_XPU_API Tensor& max_unpooling2d_forward_kernel( Tensor& output, const Tensor& self_, const Tensor& indices_, IntArrayRef output_size); -Tensor& max_unpooling3d_forward_kernel( +TORCH_XPU_API Tensor& max_unpooling3d_forward_kernel( Tensor& output, const Tensor& self_, const Tensor& indices_, diff --git a/src/ATen/native/xpu/sycl/TensorFactoriesKernels.cpp b/src/ATen/native/xpu/sycl/TensorFactoriesKernels.cpp index 23787a31b..b1b741cbf 100644 --- a/src/ATen/native/xpu/sycl/TensorFactoriesKernels.cpp +++ b/src/ATen/native/xpu/sycl/TensorFactoriesKernels.cpp @@ -1,5 +1,6 @@ #include #include +#include #include #include #include