fix: apply force cast only if at least 1 input is not quantized

sonos · Mar 1, 2024 · b8466cd · b8466cd
1 parent 739a01f
commit b8466cd
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/nnef/src/registry.rs b/nnef/src/registry.rs
@@ -227,10 +227,10 @@ impl Registry {
                     b = builder.wire_as_outlets(tract_core::ops::cast::cast(a_dt), &[b])?[0];
                     b_dt = a_dt;
                 };
+                let operating_dt = bin.1.operating_datum_type(a_dt, b_dt)?;
+                a = builder.wire_as_outlets(tract_core::ops::cast::cast(operating_dt), &[a])?[0];
+                b = builder.wire_as_outlets(tract_core::ops::cast::cast(operating_dt), &[b])?[0];
             }
-            let operating_dt = bin.1.operating_datum_type(a_dt, b_dt)?;
-            a = builder.wire_as_outlets(tract_core::ops::cast::cast(operating_dt), &[a])?[0];
-            b = builder.wire_as_outlets(tract_core::ops::cast::cast(operating_dt), &[b])?[0];
             let inputs = multi_rank_broadcast(builder, &[a, b])?;
 
             let c_dt: Option<DatumType> = dt.first().cloned().and_then(|dt| dt);