From bb2fae9bb412a22d49eea7f87170e5dcdb9812ec Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=83=91=E5=87=A0=E6=96=B9?= Date: Mon, 22 Nov 2021 17:41:41 +0800 Subject: [PATCH] python --- code/alu.py | 36 +++++++++-------- code/alu_controller.py | 89 +++++++++++++++++++++-------------------- code/binandint.py | 2 +- code/branch_unit.py | 17 ++++---- code/data_mem.py | 22 +++++----- code/forwarding_unit.py | 25 ++++++------ code/hazard_detector.py | 10 ++--- 7 files changed, 104 insertions(+), 97 deletions(-) diff --git a/code/alu.py b/code/alu.py index 6bdf879..90a6c73 100644 --- a/code/alu.py +++ b/code/alu.py @@ -1,36 +1,38 @@ import binandint -def alu(a,b,alu_ctrl): #a,b为两个运算数，alu_ctrl为控制信号，返回运算结果result与是否为0 zero - if alu_ctrl=='00000': + + +def alu(a, b, alu_ctrl): # a,b为两个运算数，alu_ctrl为控制信号，返回运算结果result与是否为0 zero + if alu_ctrl == '00000': pass - elif alu_ctrl=='00001': + elif alu_ctrl == '00001': pass - elif alu_ctrl=='00010': + elif alu_ctrl == '00010': pass - elif alu_ctrl=='00011': + elif alu_ctrl == '00011': pass - elif alu_ctrl=='00100': + elif alu_ctrl == '00100': pass - elif alu_ctrl=='00101': + elif alu_ctrl == '00101': pass - elif alu_ctrl=='00110': + elif alu_ctrl == '00110': pass - elif alu_ctrl=='00111': + elif alu_ctrl == '00111': pass - elif alu_ctrl=='01000': + elif alu_ctrl == '01000': pass - elif alu_ctrl=='01001': + elif alu_ctrl == '01001': pass - elif alu_ctrl=='01010': + elif alu_ctrl == '01010': pass - elif alu_ctrl=='01011': + elif alu_ctrl == '01011': pass - elif alu_ctrl=='01100': + elif alu_ctrl == '01100': pass - elif alu_ctrl=='01101': + elif alu_ctrl == '01101': pass - elif alu_ctrl=='01110': + elif alu_ctrl == '01110': pass - elif alu_ctrl=='01111': + elif alu_ctrl == '01111': pass else: pass \ No newline at end of file diff --git a/code/alu_controller.py b/code/alu_controller.py index a36bc80..e6466ed 100644 --- a/code/alu_controller.py +++ b/code/alu_controller.py @@ -1,48 +1,49 @@ -def alu_controller(alu_op,funct7,funct3): #alu_op来自主控制器，funct7和funct3来自指令的特定部位 返回alu_ctrl - if alu_op=='00': - opreation='00000' #lw.sw.auipc - elif alu_op=='01': - if funct3=='000': - opreation='00101' #beq - elif funct3=='001': - opreation='00110' #bne - elif funct3=='100': - opreation='00111' #blt - elif funct3=='101': - opreation='01000' #bge - elif funct3=='110': - opreation='01001' #bltu - elif funct3=='111': - opreation='01010' #bgeu +def alu_controller(alu_op, funct7, + funct3): # alu_op来自主控制器，funct7和funct3来自指令的特定部位 返回alu_ctrl + if alu_op == '00': + opreation = '00000' # lw.sw.auipc + elif alu_op == '01': + if funct3 == '000': + opreation = '00101' # beq + elif funct3 == '001': + opreation = '00110' # bne + elif funct3 == '100': + opreation = '00111' # blt + elif funct3 == '101': + opreation = '01000' # bge + elif funct3 == '110': + opreation = '01001' # bltu + elif funct3 == '111': + opreation = '01010' # bgeu else: - opreation='00000' - elif alu_op=='10': - if funct3=='000': - if funct7=='0100000': #sub - opreation='00001' - else: #add,addi - opreation='00000' - elif funct3=='100': - opreation='01100' #xor,xori - elif funct3=='110': - opreation='00010' #or,ori - elif funct3=='111': - opreation='00011' #and, andi - elif funct3=='010': - opreation='00100' #slt,slti - elif funct3=='001': - opreation='01101' #sll,slli 未存在 - elif funct3=='011': - opreation='01110' #sliu,sltiu 未存在 - elif funct3=='101': - if funct7=='0100000': #sra,srai - opreation='01111' - else: #srl,srli - opreation='10000' + opreation = '00000' + elif alu_op == '10': + if funct3 == '000': + if funct7 == '0100000': # sub + opreation = '00001' + else: # add,addi + opreation = '00000' + elif funct3 == '100': + opreation = '01100' # xor,xori + elif funct3 == '110': + opreation = '00010' # or,ori + elif funct3 == '111': + opreation = '00011' # and, andi + elif funct3 == '010': + opreation = '00100' # slt,slti + elif funct3 == '001': + opreation = '01101' # sll,slli 未存在 + elif funct3 == '011': + opreation = '01110' # sliu,sltiu 未存在 + elif funct3 == '101': + if funct7 == '0100000': # sra,srai + opreation = '01111' + else: # srl,srli + opreation = '10000' else: - opreation='00000' - elif alu_op=='11': #jal - opreation='01011' + opreation = '00000' + elif alu_op == '11': # jal + opreation = '01011' else: - opreation='00000' + opreation = '00000' return opreation diff --git a/code/binandint.py b/code/binandint.py index 389cce0..4ab5b77 100644 --- a/code/binandint.py +++ b/code/binandint.py @@ -16,4 +16,4 @@ def UInttoBin(x, n): def InttoBin(x, n): if x < 0: x += 2**n - return UInttoBin(x, n) \ No newline at end of file + return UInttoBin(x, n) diff --git a/code/branch_unit.py b/code/branch_unit.py index baf20fb..11e0ef1 100644 --- a/code/branch_unit.py +++ b/code/branch_unit.py @@ -1,9 +1,10 @@ -def branch_unit(cur_pc,imm,jalr_sel,branch_taken,alu_result): #输入值为当前pc（int），立即数，jalr信号，是否跳转信号，alu运算结果(全是int型) - pc_plus_4=cur_pc+4 #输出为pc_plus_imm,pc_plus_4,branch_target,pc_sel(忘了这是啥了，需要回头再看) - pc_plus_imm=cur_pc+imm - pc_sel = jalr_sel | (branch_taken & (alu_result%2)) - if jalr_sel==1: - branch_target=alu_result&(2**32-2) +def branch_unit(cur_pc, imm, jalr_sel, branch_taken, + alu_result): # 输入值为当前pc（int），立即数，jalr信号，是否跳转信号，alu运算结果(全是int型) + pc_plus_4 = cur_pc + 4 # 输出为pc_plus_imm,pc_plus_4,branch_target,pc_sel(忘了这是啥了，需要回头再看) + pc_plus_imm = cur_pc + imm + pc_sel = jalr_sel | (branch_taken & (alu_result % 2)) + if jalr_sel == 1: + branch_target = alu_result & (2**32 - 2) else: - branch_target=cur_pc+imm*2 - return pc_plus_imm,pc_plus_4,branch_target,pc_sel \ No newline at end of file + branch_target = cur_pc + imm * 2 + return pc_plus_imm, pc_plus_4, branch_target, pc_sel diff --git a/code/data_mem.py b/code/data_mem.py index 2846be7..b68fc90 100644 --- a/code/data_mem.py +++ b/code/data_mem.py @@ -1,26 +1,28 @@ class data_mem: - memory=[8*'0']*4*1024*16 - def mem(self,write_en,read_en,address,data_in,funct3): #写使能，读使能，(使能为int的0或1)地址，输入数据，func3 #输出为data_out + memory = [8 * '0'] * 4 * 1024 * 16 + + def mem(self, write_en, read_en, address, data_in, funct3 + ): # 写使能，读使能，(使能为int的0或1)地址，输入数据，func3 # 输出为data_out if write_en: - if funct3=='000': + if funct3 == '000': pass - elif funct3=='001': + elif funct3 == '001': pass - elif funct3=='010': + elif funct3 == '010': pass else: pass if read_en: - if funct3=='000': + if funct3 == '000': pass - elif funct3=='001': + elif funct3 == '001': pass - elif funct3=='010': + elif funct3 == '010': pass - elif funct3=='100': + elif funct3 == '100': pass - elif funct3=='101': + elif funct3 == '101': pass else: pass diff --git a/code/forwarding_unit.py b/code/forwarding_unit.py index b673390..a5cec62 100644 --- a/code/forwarding_unit.py +++ b/code/forwarding_unit.py @@ -1,14 +1,15 @@ -def fowardingunit(rs1,rs2,ex_mem_rd,mem_wb_rd,ex_mem_regwrite,mem_wb_regwrite): - if rs1!=0 and rs1==ex_mem_rd and ex_mem_regwrite: - forward_a='01' - elif rs1!=0 and rs1==mem_wb_rd and mem_wb_regwrite: - forward_a='10' +def fowardingunit(rs1, rs2, ex_mem_rd, mem_wb_rd, ex_mem_regwrite, + mem_wb_regwrite): + if rs1 != 0 and rs1 == ex_mem_rd and ex_mem_regwrite: + forward_a = '01' + elif rs1 != 0 and rs1 == mem_wb_rd and mem_wb_regwrite: + forward_a = '10' else: - forward_a='00' - if rs2!=0 and rs2==ex_mem_rd and ex_mem_regwrite: - forward_b='01' - elif rs2!=0 and rs2==mem_wb_rd and mem_wb_regwrite: - forward_b='10' + forward_a = '00' + if rs2 != 0 and rs2 == ex_mem_rd and ex_mem_regwrite: + forward_b = '01' + elif rs2 != 0 and rs2 == mem_wb_rd and mem_wb_regwrite: + forward_b = '10' else: - forward_b='00' - return forward_a,forward_b + forward_b = '00' + return forward_a, forward_b diff --git a/code/hazard_detector.py b/code/hazard_detector.py index 167110c..1e53c48 100644 --- a/code/hazard_detector.py +++ b/code/hazard_detector.py @@ -1,6 +1,6 @@ -def hazard_detector(if_id_rs1,if_id_rs2,id_ex_rd,id_ex_memread): - if id_ex_memread and (id_ex_rd==if_id_rs1 or id_ex_rd==if_id_rs2): - stall=1 +def hazard_detector(if_id_rs1, if_id_rs2, id_ex_rd, id_ex_memread): + if id_ex_memread and (id_ex_rd == if_id_rs1 or id_ex_rd == if_id_rs2): + stall = 1 else: - stall=0 - return stall \ No newline at end of file + stall = 0 + return stall