Mesa (main): rusticl/kernel: explicitly set rounding modes

[GitLab Mirror]

Module: Mesa
Branch: main
Commit: d136583c82d5036696b322bbe966490b321c96e1
URL:    
http://cgit.freedesktop.org/mesa/mesa/commit/?id=d136583c82d5036696b322bbe966490b321c96e1

Author: Karol Herbst <kher...@redhat.com>
Date:   Tue Dec 12 15:46:48 2023 +0100

rusticl/kernel: explicitly set rounding modes

Since dbbf566588c ("aco,ac/llvm,radeonsi: lower f2f16 to f2f16_rtz in nir")
radeonsi behavior changed and some of the core fp16 ops broke as a result.

We should explicitly specify the rounding mode until we add an gallium API
for drivers to advertize what they prefer.

Cc: mesa-stable
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/26655>

---

 src/gallium/frontends/rusticl/core/kernel.rs       | 4 ++++
 src/gallium/frontends/rusticl/mesa/compiler/nir.rs | 9 +++++++++
 2 files changed, 13 insertions(+)

diff --git a/src/gallium/frontends/rusticl/core/kernel.rs 
b/src/gallium/frontends/rusticl/core/kernel.rs
index 3d5b724efea..3b116bf68a0 100644
--- a/src/gallium/frontends/rusticl/core/kernel.rs
+++ b/src/gallium/frontends/rusticl/core/kernel.rs
@@ -734,6 +734,10 @@ pub(super) fn convert_spirv_to_nir(
          */
         nir.preserve_fp16_denorms();
 
+        // Set to rtne for now until drivers are able to report their prefered 
rounding mode, that
+        // also matches what we report via the API.
+        nir.set_fp_rounding_mode_rtne();
+
         let (args, internal_args) = lower_and_optimize_nir(dev, &mut nir, 
args, &dev.lib_clc);
 
         if let Some(cache) = cache {
diff --git a/src/gallium/frontends/rusticl/mesa/compiler/nir.rs 
b/src/gallium/frontends/rusticl/mesa/compiler/nir.rs
index bb5fafdc261..b61ec9254a9 100644
--- a/src/gallium/frontends/rusticl/mesa/compiler/nir.rs
+++ b/src/gallium/frontends/rusticl/mesa/compiler/nir.rs
@@ -446,6 +446,15 @@ impl NirShader {
         }
     }
 
+    pub fn set_fp_rounding_mode_rtne(&mut self) {
+        unsafe {
+            self.nir.as_mut().info.float_controls_execution_mode |=
+                float_controls::FLOAT_CONTROLS_ROUNDING_MODE_RTE_FP16 as u32
+                    | float_controls::FLOAT_CONTROLS_ROUNDING_MODE_RTE_FP32 as 
u32
+                    | float_controls::FLOAT_CONTROLS_ROUNDING_MODE_RTE_FP64 as 
u32;
+        }
+    }
+
     pub fn reads_sysval(&self, sysval: gl_system_value) -> bool {
         let nir = unsafe { self.nir.as_ref() };
         bitset::test_bit(&nir.info.system_values_read, sysval as u32)