From b14ca7141e144ef46642917ac7845216969412bb Mon Sep 17 00:00:00 2001
From: Yujie Seah <155465254+boringbox@users.noreply.github.com>
Date: Mon, 21 Sep 2026 03:35:36 +0000
Subject: [PATCH 1/2] I have instantiated the zba extension to the exectute
 stage of the cpu. I also made some ' changes to the zba extension as part of
 this process.g

---
 src/ext/b/zba.sv            | 10 ++++------
 src/stages/execute_stage.sv |  8 +++++++-
 2 files changed, 11 insertions(+), 7 deletions(-)

diff --git a/src/ext/b/zba.sv b/src/ext/b/zba.sv
index 63b4503d..6df67882 100644
--- a/src/ext/b/zba.sv
+++ b/src/ext/b/zba.sv
@@ -9,10 +9,8 @@
 
 module zba(input [31:0] reg1
   , input [31:0] reg2
-  , input [1:0] inst // two bit register, distinguish instruction
+  , input [2:0] inst // two bit register, distinguish instruction
   /* verilator lint_off UNUSEDSIGNAL */
-  , input [2:0] funct3  //extra function inputs
-  , input [6:0] funct7
   , output logic[31:0] out
   );
 
@@ -22,15 +20,15 @@ always_comb
 
 // sh1add behaviour
 
-    2'b00: out = (reg1 << 1) + reg2;
+    3'b010: out = (reg1 << 1) + reg2;
 
 // sh2add behaviour
 
-    2'b01: out = (reg1 << 2) + reg2;
+    3'b100: out = (reg1 << 2) + reg2;
 
 // sh3add
 
-    2'b10: out = (reg1 << 3) + reg2;
+    3'b110: out = (reg1 << 3) + reg2;
 
 // other
 
diff --git a/src/stages/execute_stage.sv b/src/stages/execute_stage.sv
index ffe17882..94d4f165 100644
--- a/src/stages/execute_stage.sv
+++ b/src/stages/execute_stage.sv
@@ -25,7 +25,7 @@ module execute_stage
   logic zero_flag;
 
 `ifdef UTOSS_RISCV_ENABLE_B_EXT
-  data_t zbb_result, zbkb_result;
+  data_t zba_result, zbb_result, zbkb_result;
   logic  zbb_zero_flag, zbkb_zero_flag;
 `endif
 
@@ -73,6 +73,12 @@ module execute_stage
     );
 
 `ifdef UTOSS_RISCV_ENABLE_B_EXT
+  zba u_zba
+    ( .reg1 (rs1)
+    , .reg2 (rs2)
+    , .inst (id_to_ex.funct3)
+    , .out (zba_result)
+    );
   zbb u_zbb
     ( .a              ( alu_input_a            )
     , .b              ( alu_input_b            )

From 0517fb63f64025f8aa243973f495daded24b3865 Mon Sep 17 00:00:00 2001
From: "Dazhou H." <118654438+Dazhou-20383@users.noreply.github.com>
Date: Sun, 27 Sep 2026 15:09:06 -0400
Subject: [PATCH 2/2] Fix comment for inst input size in zba module

---
 src/ext/b/zba.sv | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/ext/b/zba.sv b/src/ext/b/zba.sv
index 6df67882..d16e78ed 100644
--- a/src/ext/b/zba.sv
+++ b/src/ext/b/zba.sv
@@ -9,7 +9,7 @@
 
 module zba(input [31:0] reg1
   , input [31:0] reg2
-  , input [2:0] inst // two bit register, distinguish instruction
+  , input [2:0] inst // three bit register, distinguish instruction
   /* verilator lint_off UNUSEDSIGNAL */
   , output logic[31:0] out
   );