From d77ec3dcac4032d8f051ffc45131a0a69e3c2c41 Mon Sep 17 00:00:00 2001 From: Martin Robinson Date: Mon, 27 Jan 2025 21:32:26 +0000 Subject: [PATCH] bug: remove rhs and lhs from data layout (#46) --- src/execution/data_layout.rs | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/execution/data_layout.rs b/src/execution/data_layout.rs index f52e690..ba53496 100644 --- a/src/execution/data_layout.rs +++ b/src/execution/data_layout.rs @@ -35,9 +35,10 @@ impl DataLayout { let mut add_tensor = |tensor: &Tensor| { let is_state = tensor.name() == "u" || tensor.name() == "dudt"; + let is_rhs_or_lhs = tensor.name() == "rhs" || tensor.name() == "lhs"; // insert the data (non-zeros) for each tensor layout_map.insert(tensor.name().to_string(), tensor.layout_ptr().clone()); - if !is_state { + if !is_state && !is_rhs_or_lhs { data_index_map.insert(tensor.name().to_string(), data.len()); data_length_map.insert(tensor.name().to_string(), tensor.nnz()); data.extend(vec![0.0; tensor.nnz()]);