MatMul.S 2.46 KB
Newer Older
YouGuoliang's avatar
YouGuoliang committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
# 伪矩阵乘法  汇编代码
# 我们的 RV32I CPU 没有实现乘法指令,所以在伪矩阵乘法中,使用按位或代替加法,用加法代替乘法,完成矩阵运算。
# 虽然不是真的矩阵乘法,但能够模仿矩阵乘法对RAM的访问过程,对cache的性能研究起到作用
#

.org 0x0
    .global _start
_start:
    xori   a4, zero, 4       # a4寄存器决定了计算的规模,矩阵规模=N*NN=2^a4。例如a4=4,则矩阵为 2^4=16阶方阵。该值可以修改。当然,矩阵规模变化后,DataRam的内存分配方式也要同步的变化,才能运行出正确结果

    # 以下指令计算3个矩阵(目的矩阵,源矩阵1,源矩阵2)在内存中的起始地址。
    # 这三个矩阵在内存中顺序而紧挨着存放,例如 a4=4,则N=16,则每个矩阵占N*N=256个字,即1024个字节
    #  目的矩阵起始地址为0  源矩阵1起始地址为1024  源矩阵2起始地址为2048
    # 目的矩阵起始地址放在a2里,源矩阵1起始地址放在a0里,源矩阵2起始地址放在a1
    xori   a3, zero, 4
    sll    a3, a3  , a4
    xor    a2, zero, zero
    sll    a0, a3  , a4
    add    a1, a0  , a0
    
    # 开始矩阵乘法,使用伪矩阵乘法公式:c_{ij} = \sigma c_{ik}*b{kj}  循环嵌套顺序(从内向外)为 i,j,k  分别使用 t0,t1,t2 存放 i,j,k
    xor    t0, zero, zero
    MatMulLoopI:
        xor    t1, zero, zero
        MatMulLoopJ:
            xor    t3, zero, zero    #t3存放最内求和循环的累加和,首先将t3清零
            xor    t2, zero, zero
            MatMulLoopK:
                sll  t4, t0, a4
                add  t4, t4, t2
                add  t4, t4, a0
                lw   t4, (t4)
                sll  t5, t2, a4
                add  t5, t5, t1
                add  t5, t5, a1
                lw   t5, (t5)
                and  t4, t4, t5
                add  t3, t3, t4
                addi t2, t2, 4
            blt    t2, a3, MatMulLoopK
            sll    t4, t0, a4
            add    t4, t4, t1
            add    t4, t4, a2
            sw     t3, (t4)
            addi   t1, t1, 4
        blt    t1, a3, MatMulLoopJ
        addi   t0, t0, 4
    blt    t0, a3, MatMulLoopI
    




    add    t0, zero, 1
    sll    t0, t0, a4
    sll    t0, t0, a4
    slli   t0, t0, 2
    addi   t1, zero, 0
Loop:
    lw     t2, (t1)
    addi   t1, t1, 4
    blt    t1, t0, Loop
# 计算结束,死循环
InfLoop:
    jal    zero, InfLoop