rust - iterate over qf input/output size exactly

CEED · Feb 10, 2024 · 58f118f · 58f118f
1 parent 31b0218
commit 58f118f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rust/libceed/src/qfunction.rs b/rust/libceed/src/qfunction.rs
@@ -541,6 +541,7 @@ unsafe extern "C" fn trampoline(
     let mut inputs_array: [&[crate::Scalar]; MAX_QFUNCTION_FIELDS] = [&[0.0]; MAX_QFUNCTION_FIELDS];
     inputs_slice
         .iter()
+        .take(trampoline_data.number_inputs)
         .enumerate()
         .map(|(i, &x)| {
             std::slice::from_raw_parts(x, trampoline_data.input_sizes[i] * q as usize)
@@ -556,6 +557,7 @@ unsafe extern "C" fn trampoline(
         mut_max_fields!(&mut [0.0]);
     outputs_slice
         .iter()
+        .take(trampoline_data.number_outputs)
         .enumerate()
         .map(|(i, &x)| {
             std::slice::from_raw_parts_mut(x, trampoline_data.output_sizes[i] * q as usize)