fix: index error for total memory chunks

root-project · Jan 29, 2025 · b42e49a · b42e49a
1 parent f5f0dfa
commit b42e49a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tmva/sofie/src/RModel.cxx b/tmva/sofie/src/RModel.cxx
@@ -374,12 +374,13 @@ std::string RModel::AllocateIntermediateMemory(std::span<const std::string_view>
             if (fIntermediateMemoryInfo.total_memory.size()){
                auto previous_tensor_idx = fIntermediateMemoryInfo.total_memory.back().chunk_idx;
                auto previous_tensor_size = fIntermediateMemoryInfo.total_memory.back().tensor_size;
+
                fIntermediateMemoryInfo.total_memory.push_back({
                   std::string(it),
                   previous_tensor_idx+previous_tensor_size,
                   tensor_size
                });
-               memory_allocation_string += "\nfloat* tensor_"+ std::string(it) + "= reinterpret_cast<float*>(fIntermediateMemoryPool+" + previous_tensor_idx+previous_tensor_size + ");\n";
+               memory_allocation_string += "\nfloat* tensor_"+ std::string(it) + "= reinterpret_cast<float*>(fIntermediateMemoryPool+" + std::to_string(previous_tensor_idx+previous_tensor_size) + ");\n";
             } else {
                fIntermediateMemoryInfo.total_memory.push_back({
                   std::string(it),