tiny.simd.BFloat16

1pub const BFloat16 = extern struct {2    pub const is_bfloat16 = true;3    pub const Storage = u16;45    bits: u16,67    pub fn fromBits(bits: u16) BFloat16 {8        return .{ .bits = bits };9    }1011    pub fn fromF32(value: f32) BFloat16 {12        return .{ .bits = bitsFromF32(value) };13    }1415    pub fn fromF64(value: f64) BFloat16 {16        const bits: u64 = @bitCast(value);17        const rounded_bits = (bits & 0xffff_ffc0_0000_0000) |18            ((bits +% 0x0000_003f_ffff_ffff) & 0x0000_0040_0000_0000);19        return fromF32(@floatCast(@as(f64, @bitCast(rounded_bits))));20    }2122    pub fn toF32(self: BFloat16) f32 {23        return f32FromBits(self.bits);24    }2526    pub fn neg(self: BFloat16) BFloat16 {27        return fromBits(self.bits ^ 0x8000);28    }2930    pub fn add(self: BFloat16, other: BFloat16) BFloat16 {31        return fromF32(self.toF32() + other.toF32());32    }3334    pub fn sub(self: BFloat16, other: BFloat16) BFloat16 {35        return fromF32(self.toF32() - other.toF32());36    }3738    pub fn mul(self: BFloat16, other: BFloat16) BFloat16 {39        return fromF32(self.toF32() * other.toF32());40    }4142    pub fn div(self: BFloat16, other: BFloat16) BFloat16 {43        return fromF32(self.toF32() / other.toF32());44    }45};

1pub const BFloat16 = bfloat.BFloat16;