Fix ptr initialization

intel · Jun 20, 2024 · fc5014d · fc5014d
1 parent 989eb44
commit fc5014d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/aten/sycl/AveragePool2dKernels.cpp b/src/aten/sycl/AveragePool2dKernels.cpp
@@ -243,8 +243,8 @@ void avg_pool2d_channels_last_frame(
     const int divisor_override,
     const bool count_include_pad,
     const bool use_divisor) {
-  scalar_t* top_data = input.data_ptr<scalar_t>();
-  const scalar_t* bottom_data = output.data_ptr<scalar_t>();
+  scalar_t* top_data = output.data_ptr<scalar_t>();
+  const scalar_t* bottom_data = input.data_ptr<scalar_t>();
 
   auto& queue = at::xpu::getCurrentSYCLQueue();
   const uint32_t group_size =