1 files changed, 18 insertions, 0 deletions
diff --git a/src/shift_unit.v b/src/shift_unit.v
new file mode 100644
index 0000000..55b6add
--- /dev/null
+++ b/src/shift_unit.v
@@ -0,0 +1,18 @@
+module shift_unit #(
+  parameter N = 32
+)(
+  input [N-1:0] A,
+  input [clog2(N):0] SHAMT,
+  input [1:0] OP, // 00: SLL, 01: SRL, 11: SRA 
+  output reg [N-1:0] RESULT
+);
+
+always @ (*) begin
+  case (OP)
+    2'b00: RESULT <= A << SHAMT;
+    2'b01: RESULT <= A >> SHAMT;
+    2'b11: RESULT <= A >>> SHAMT;
+  endcase
+end
+
+endmodule