Add TF32

This floating point type is used by some accelerators (e.g., AMD and NVIDIA GPUs)

Upstream LLVM commit: 55c2211a233e11179048cf58778f40e5a62f444a

Author: eddyb

Cargo.toml

@@ -2,7 +2,7 @@
 members = ["fuzz"]
 
 [workspace.package]
-version = "0.2.0+llvm-687bd77e2c26"
+version = "0.2.0+llvm-55c2211a233e"
 edition = "2021"
 license = "Apache-2.0 WITH LLVM-exception"
 

fuzz/ops.rs

@@ -331,6 +331,7 @@ struct FuzzOp {
         (8, "Float8E4M3FNUZ"),
         (8, "Float8E4M3B11FNUZ"),
         (16, "BFloat"),
+        (19, "FloatTF32"),
         (80, "x87DoubleExtended"),
     ]
     .into_iter()
@@ -339,6 +340,7 @@ struct FuzzOp {
         let name = match (w, cxx_apf_semantics) {
             (8, s) if s.starts_with("Float8") => s.replace("Float8", "F8"),
             (16, "BFloat") => "BrainF16".into(),
+            (19, "FloatTF32") => "NV_TensorF32".into(),
             (80, "x87DoubleExtended") => "X87_F80".into(),
             _ => {
                 assert!(cxx_apf_semantics.starts_with("IEEE"));

fuzz/src/main.rs

@@ -263,6 +263,11 @@ float_reprs! {
         const REPR_TAG = 16 + 1;
         extern fn = cxx_apf_fuzz_eval_op_brainf16;
     }
+    NV_TensorF32(u32) {
+        type RustcApFloat = rustc_apfloat::ieee::FloatTF32;
+        const REPR_TAG = 32 + 1;
+        extern fn = cxx_apf_fuzz_eval_op_nv_tensorf32;
+    }
     X87_F80(u128) {
         type RustcApFloat = rustc_apfloat::ieee::X87DoubleExtended;
         extern fn = cxx_apf_fuzz_eval_op_x87_f80;

src/ieee.rs

@@ -386,6 +386,11 @@ ieee_semantics! {
         const NAN_ENCODING: NanEncoding = NanEncoding::NegativeZero;
         const MIN_EXP: ExpInt = Self::IEEE_MIN_EXP - 4;
     },
+
+    // Floating point number that occupies 32 bits or less of storage, providing
+    // improved range compared to half (16-bit) formats, at (potentially)
+    // greater throughput than single precision (32-bit) formats.
+    FloatTF32 = FloatTF32S(19:8),
 }
 
 // FIXME(eddyb) consider moving X87-specific logic to a "has explicit integer bit"

src/lib.rs

@@ -1,6 +1,6 @@
 //! Port of LLVM's APFloat software floating-point implementation from the
 //! following C++ sources (please update commit hash when backporting):
-//! https://github.com/llvm/llvm-project/commit/687bd77e2c26487cba727aacfa7067dd01286be0
+//! https://github.com/llvm/llvm-project/commit/55c2211a233e11179048cf58778f40e5a62f444a
 //! * `llvm/include/llvm/ADT/APFloat.h` -> `Float` and `FloatConvert` traits
 //! * `llvm/lib/Support/APFloat.cpp` -> `ieee` and `ppc` modules
 //! * `llvm/unittests/ADT/APFloatTest.cpp` -> `tests` directory