AsPrimitive f16 <-> bf16

starkat99 · Feb 22, 2024 · 3223add · 3223add
1 parent a513d3b
commit 3223add
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/num_traits.rs b/src/num_traits.rs
@@ -718,6 +718,7 @@ impl_as_primitive_to_f16!(isize, to_f32);
 impl_as_primitive_to_f16!(usize, to_f32);
 impl_as_primitive_to_f16!(f32, to_f32);
 impl_as_primitive_to_f16!(f64, to_f64);
+impl_as_primitive_to_f16!(bf16, to_f32);
 
 macro_rules! impl_as_primitive_f16_from {
     ($ty:ty, $meth:ident) => {
@@ -1457,6 +1458,7 @@ impl_as_primitive_to_bf16!(isize, to_f32);
 impl_as_primitive_to_bf16!(usize, to_f32);
 impl_as_primitive_to_bf16!(f32, to_f32);
 impl_as_primitive_to_bf16!(f64, to_f64);
+impl_as_primitive_to_bf16!(f16, to_f32);
 
 macro_rules! impl_as_primitive_bf16_from {
     ($ty:ty, $meth:ident) => {