From f919752870dd78ccab546a5a0990ff24fbc94055 Mon Sep 17 00:00:00 2001 From: Feng Yuan Date: Sat, 13 Jul 2024 00:18:39 +0800 Subject: [PATCH] Fix compilation error --- .../native/xpu/sycl/ActivationLogSigmoidKernels.cpp | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/src/ATen/native/xpu/sycl/ActivationLogSigmoidKernels.cpp b/src/ATen/native/xpu/sycl/ActivationLogSigmoidKernels.cpp index 6e21b109f..fcb44b439 100644 --- a/src/ATen/native/xpu/sycl/ActivationLogSigmoidKernels.cpp +++ b/src/ATen/native/xpu/sycl/ActivationLogSigmoidKernels.cpp @@ -35,10 +35,10 @@ struct LogSigmoidBackwardFunctor { const opmath_t in = in_; const opmath_t grad_out = grad_out_; - auto in_negative = in < opmath_t(0); - auto max_deriv = in_negative ? opmath_t(1) : opmath_t(0); - auto sign = in_negative ? opmath_t(1) : -opmath_t(1); - const auto z = std::exp(-std::abs(in)); + bool in_negative = in < opmath_t(0); + opmath_t max_deriv = in_negative ? opmath_t(1) : opmath_t(0); + opmath_t sign = in_negative ? opmath_t(1) : -opmath_t(1); + const opmath_t z = std::exp(-std::abs(in)); return grad_out * (max_deriv - sign * (z / (opmath_t(1) + z))); } }; @@ -52,4 +52,4 @@ void log_sigmoid_backward_kernel(TensorIteratorBase& iter) { [&]() { gpu_kernel(iter, LogSigmoidBackwardFunctor()); }); } -} // namespace at::native::xpu \ No newline at end of file +} // namespace at::native::xpu