Windows: Fixing building error (#1019)

Signed-off-by: Feng Yuan <[email protected]>
intel · Oct 28, 2024 · bdbda35 · bdbda35
1 parent 69ebbed
commit bdbda35
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -1,5 +1,6 @@
 #include <ATen/native/TensorFactories.h>
 #include <ATen/native/xpu/sycl/ScanUtils.h>
+#include <ATen/native/xpu/sycl/TensorFactoriesKernels.h>
 #include <ATen/xpu/EmptyTensor.h>
 #include <c10/core/TensorOptions.h>
 #include <comm/DeviceProperties.h>