Merge pull request #101 from charles-r-earp/as_primitive_f16_to_bf16

AsPrimitive f16 <-> bf16
starkat99 · Feb 26, 2024 · 7cf20dc · 7cf20dc
2 parents 1b59e9e + 3223add
commit 7cf20dc
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) => {