From d4c243c9ef1679b1e463ed2d22ffa7153038a1d4 Mon Sep 17 00:00:00 2001
From: Daniel Maddern <daniel.maddern@illuvium.io>
Date: Sat, 15 Aug 2026 02:48:06 +0700
Subject: [PATCH] Match NVFP4 tie-to-even rounding

---
 src/h3_blackwell_runtime/csrc/nvfp4_scale.cu | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/h3_blackwell_runtime/csrc/nvfp4_scale.cu b/src/h3_blackwell_runtime/csrc/nvfp4_scale.cu
index 6fb519c..630fa4c 100644
--- a/src/h3_blackwell_runtime/csrc/nvfp4_scale.cu
+++ b/src/h3_blackwell_runtime/csrc/nvfp4_scale.cu
@@ -60,15 +60,15 @@ __inline__ __device__ uint8_t encode_fp4_e2m1(float value) {
   uint8_t code = 0;
   if (abs_value > 5.0f) {
     code = 7;
-  } else if (abs_value > 3.5f) {
+  } else if (abs_value >= 3.5f) {
     code = 6;
   } else if (abs_value > 2.5f) {
     code = 5;
-  } else if (abs_value > 1.75f) {
+  } else if (abs_value >= 1.75f) {
     code = 4;
   } else if (abs_value > 1.25f) {
     code = 3;
-  } else if (abs_value > 0.75f) {
+  } else if (abs_value >= 0.75f) {
     code = 2;
   } else if (abs_value > 0.25f) {
     code = 1;