AsPrimitive f16 <-> bf16

starkat99 · Jan 13, 2024 · 9734f04 · 9734f04
1 parent cabfc74
commit 9734f04
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) => {