Merge pull request #65 from tracel-ai/fix/wgpu-float

Revert float precision changes to wgsl compiler
tracel-ai · Aug 12, 2024 · 546f588 · 546f588
2 parents c3ef475 + 4aa75fd
commit 546f588
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/crates/cubecl-wgpu/src/compiler/wgsl/base.rs b/crates/cubecl-wgpu/src/compiler/wgsl/base.rs
@@ -259,8 +259,8 @@ impl Display for Variable {
                     FloatKind::BF16 => {
                         todo!("Unsupported")
                     }
-                    FloatKind::F32 => f.write_fmt(format_args!("{:.9}f", *val as f32)),
-                    FloatKind::F64 => f.write_fmt(format_args!("{:.17}f", { *val })),
+                    FloatKind::F32 => f.write_fmt(format_args!("{}f", *val as f32)),
+                    FloatKind::F64 => f.write_fmt(format_args!("{}f", { *val })),
                 },
                 ConstantScalarValue::UInt(val) => f.write_fmt(format_args!("{}u", *val as u32)),
                 ConstantScalarValue::Bool(val) => f.write_fmt(format_args!("{}", val)),