Fix bug where too many vector elements were written back

1 files changed, 8 insertions, 1 deletions

diff --git a/inc/wb.h b/inc/wb.h
index d3a1b93..35c9240 100644
--- a/inc/wb.h
+++ b/inc/wb.h
@@ -24,8 +24,8 @@
 class WB : public Stage
 {
   public:
-	using Stage::Stage;
 	using Stage::advance;
+	using Stage::Stage;
 
   private:
 	void advance_helper() override;
@@ -47,6 +47,13 @@ class WB : public Stage
 	 * STORE.
 	 */
 	bool should_jump();
+	/**
+	 * @return the vector register to be stored, obtained by copying the
+	 * unfilled elements in the destination register to the source. This is
+	 * required to ensure what is written back only changes VECTOR_LENGTH number
+	 * of elements.
+	 */
+	std::array<signed int, V_R_LIMIT> copy_extra_vector_elements();
 };
 
 #endif /* WB_H_INCLUDED */