@@ -96,27 +96,47 @@ CGEMVTKERNEL = zgemv_t_rvv.c
96
96
ZGEMVTKERNEL = zgemv_t_rvv.c
97
97
98
98
SGEMMKERNEL = sgemm_kernel_$(SGEMM_UNROLL_M)x$(SGEMM_UNROLL_N)_zvl128b.c
99
+ ifneq ($(filter $(SGEMM_UNROLL_N),4 8 16),)
100
+ SGEMMONCOPY = gemm_ncopy_$(SGEMM_UNROLL_N)_rvv.c
101
+ SGEMMOTCOPY = gemm_tcopy_$(SGEMM_UNROLL_N)_rvv.c
102
+ else
99
103
SGEMMONCOPY = ../generic/gemm_ncopy_$(SGEMM_UNROLL_N).c
100
104
SGEMMOTCOPY = ../generic/gemm_tcopy_$(SGEMM_UNROLL_N).c
105
+ endif
101
106
SGEMMONCOPYOBJ = sgemm_oncopy$(TSUFFIX).$(SUFFIX)
102
107
SGEMMOTCOPYOBJ = sgemm_otcopy$(TSUFFIX).$(SUFFIX)
103
108
104
109
ifneq ($(SGEMM_UNROLL_M), $(SGEMM_UNROLL_N))
110
+ ifneq ($(filter $(SGEMM_UNROLL_M),4 8 16),)
111
+ SGEMMINCOPY = gemm_ncopy_$(SGEMM_UNROLL_M)_rvv.c
112
+ SGEMMITCOPY = gemm_tcopy_$(SGEMM_UNROLL_M)_rvv.c
113
+ else
105
114
SGEMMINCOPY = ../generic/gemm_ncopy_$(SGEMM_UNROLL_M).c
106
115
SGEMMITCOPY = ../generic/gemm_tcopy_$(SGEMM_UNROLL_M).c
116
+ endif
107
117
SGEMMINCOPYOBJ = sgemm_incopy$(TSUFFIX).$(SUFFIX)
108
118
SGEMMITCOPYOBJ = sgemm_itcopy$(TSUFFIX).$(SUFFIX)
109
119
endif
110
120
111
121
DGEMMKERNEL = dgemm_kernel_$(DGEMM_UNROLL_M)x$(DGEMM_UNROLL_N)_zvl128b.c
122
+ ifneq ($(filter $(DGEMM_UNROLL_N),4 8 16),)
123
+ DGEMMONCOPY = gemm_ncopy_$(DGEMM_UNROLL_N)_rvv.c
124
+ DGEMMOTCOPY = gemm_tcopy_$(DGEMM_UNROLL_N)_rvv.c
125
+ else
112
126
DGEMMONCOPY = ../generic/gemm_ncopy_$(DGEMM_UNROLL_N).c
113
127
DGEMMOTCOPY = ../generic/gemm_tcopy_$(DGEMM_UNROLL_N).c
128
+ endif
114
129
DGEMMONCOPYOBJ = dgemm_oncopy$(TSUFFIX).$(SUFFIX)
115
130
DGEMMOTCOPYOBJ = dgemm_otcopy$(TSUFFIX).$(SUFFIX)
116
131
117
132
ifneq ($(DGEMM_UNROLL_M), $(DGEMM_UNROLL_N))
133
+ ifneq ($(filter $(DGEMM_UNROLL_M),4 8 16),)
134
+ DGEMMINCOPY = gemm_ncopy_$(DGEMM_UNROLL_M)_rvv.c
135
+ DGEMMITCOPY = gemm_tcopy_$(DGEMM_UNROLL_M)_rvv.c
136
+ else
118
137
DGEMMINCOPY = ../generic/gemm_ncopy_$(DGEMM_UNROLL_M).c
119
138
DGEMMITCOPY = ../generic/gemm_tcopy_$(DGEMM_UNROLL_M).c
139
+ endif
120
140
DGEMMINCOPYOBJ = dgemm_incopy$(TSUFFIX).$(SUFFIX)
121
141
DGEMMITCOPYOBJ = dgemm_itcopy$(TSUFFIX).$(SUFFIX)
122
142
endif
@@ -255,4 +275,4 @@ SHGEMMOTCOPYOBJ = shgemm_otcopy$(TSUFFIX).$(SUFFIX)
255
275
ifndef SHGEMM_BETA
256
276
SHGEMM_BETA = gemm_beta_rvv.c
257
277
endif
258
- endif
278
+ endif
0 commit comments