From 9956cfeba694c923736da047cc5ef8171dfff0f5 Mon Sep 17 00:00:00 2001 From: "Gao, Xiang" Date: Fri, 25 Jul 2025 10:41:09 -0700 Subject: [PATCH] Fix test_correctness_abs_float64 --- runtime/helpers.cu | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/runtime/helpers.cu b/runtime/helpers.cu index b343d1225ea..37d59595625 100644 --- a/runtime/helpers.cu +++ b/runtime/helpers.cu @@ -114,6 +114,10 @@ __device__ __bfloat fmax(__bfloat a, __bfloat b) { return __bfloat2float(a) > __bfloat2float(b) ? a : b; } +__device__ double abs(double a) { + return a < 0 ? -a : a; +} + __device__ float abs(float a) { return fabs(a); }