1 files changed, 21 insertions, 0 deletions
diff --git a/rtl/src/shift_unit.v b/rtl/src/shift_unit.v
new file mode 100644
index 0000000..ea83e4a
--- /dev/null
+++ b/rtl/src/shift_unit.v
@@ -0,0 +1,21 @@
+module shift_unit (
+  input signed [31:0] a,
+  input  [4:0] b,
+
+  input [1:0] op,
+  
+  output reg [31:0] result
+);
+
+`include "include/consts.vh"
+
+always @ (*) begin
+  case (op)
+    SHIFT_OP_SLL: result = a << b;   // SLL
+    SHIFT_OP_SRL: result = a >> b;   // SRL
+    SHIFT_OP_SRA: result = a >>> b;  // SRA
+    default:      result = 32'b0;
+  endcase
+end
+
+endmodule