Fix sampling bug in RoiAlign

intel · Jan 17, 2025 · 8a8ee8f · 8a8ee8f
1 parent 0274bf9
commit 8a8ee8f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/ATen/native/xpu/sycl/RoiAlignKernels.cpp b/src/ATen/native/xpu/sycl/RoiAlignKernels.cpp
@@ -156,6 +156,7 @@ struct RoiAlignForwardKernel {
         width_(width),
         pooled_height_(pooled_height),
         pooled_width_(pooled_width),
+        sampling_ratio_(sampling_ratio),
         aligned_(aligned),
         rois_(rois),
         output_(output) {}