Update UpSampleBilinear2d.cpp

intel · Nov 29, 2024 · 5b30cae · 5b30cae
1 parent c9d0183
commit 5b30cae
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/ATen/native/xpu/UpSampleBilinear2d.cpp b/src/ATen/native/xpu/UpSampleBilinear2d.cpp
@@ -30,6 +30,7 @@ TORCH_IMPL_FUNC(upsample_bilinear2d_backward_out_xpu)
  std::optional<double> scales_h,
  std::optional<double> scales_w,
  const Tensor& grad_input) {
+  globalContext().alertNotDeterministic("upsample_bilinear2d_backward_out_xpu");
   xpu::upsample_bilinear2d_backward_out_kernel(
       grad_input,
       grad_output,