Merge pull request #1359 from sonos/task/improve-f32-to-f16

Add cast op conversion from f32 to f16

core/src/floats.rs

@@ -4,6 +4,7 @@ use crate::internal::translator::Translate;
 use crate::internal::*;
 use crate::ops::array::{Pad, PadMode};
 use crate::ops::einsum::EinSum;
+use crate::ops::cast::Cast;
 use crate::ops::konst::Const;
 use crate::ops::scan::Scan;
 use crate::ops::source::TypedSource;
@@ -39,6 +40,12 @@ impl<T1: Datum + Float, T2: Datum + Float>
             Box::new(TypedSource::new(fact_float_precision_conversion::<T1, T2>(&source.fact)))
         } else if let Some(konst) = node.op_as::<Const>() {
             Box::new(Const(tensor_float_precision_conversion::<T1, T2>(&konst.0)))
+        } else if let Some(cast) = node.op_as::<Cast>() {
+            if cast.to == T1::datum_type() {
+                Box::new(Cast { to: T2::datum_type() })
+            } else {
+                node.op.clone()
+            }
         } else if let Some(op) = node.op_as::<Scan>() {
             let body = FloatPrecisionTranslator::<T1, T2>::default().translate_model(&op.body)?;
             Box::new(Scan { body, ..op.clone() })