diff --git a/aten/src/ATen/native/cuda/UnarySignKernels.cu b/aten/src/ATen/native/cuda/UnarySignKernels.cu index 7e74f05649da..2d812c1c8304 100644 --- a/aten/src/ATen/native/cuda/UnarySignKernels.cu +++ b/aten/src/ATen/native/cuda/UnarySignKernels.cu @@ -78,7 +78,7 @@ void signbit_kernel_cuda(TensorIteratorBase& iter){ if (at::isIntegralType(iter.input_dtype(), /*includeBool=*/false)) { AT_DISPATCH_INTEGRAL_TYPES(iter.input_dtype(), "signbit_cuda", [&]() { using opmath_t = at::opmath_type; - gpu_kernel(iter, []GPU_LAMBDA(scalar_t a) -> bool { return a < 0; }); + gpu_kernel(iter, []GPU_LAMBDA(scalar_t a) -> bool { return is_negative(a); }); }); } else { AT_DISPATCH_FLOATING_TYPES_AND2(kBFloat16, ScalarType::Half, iter.input_dtype(), "signbit_cuda", [&]() {