Skip to content

Commit 2847354

Browse files
committed
Allow target-specific GEMM and TRSM overrides
1 parent 70faa9f commit 2847354

1 file changed

Lines changed: 36 additions & 0 deletions

File tree

kernel/wasm/KERNEL

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -100,17 +100,37 @@ DTRMMKERNEL = ../generic/trmmkernel_2x2.c
100100
CTRMMKERNEL = ../generic/ztrmmkernel_2x2.c
101101
ZTRMMKERNEL = ../generic/ztrmmkernel_2x2.c
102102

103+
ifndef SGEMMKERNEL
103104
SGEMMKERNEL = ../generic/gemmkernel_2x2.c
105+
endif
106+
ifndef SGEMMONCOPY
104107
SGEMMONCOPY = ../generic/gemm_ncopy_2.c
108+
endif
109+
ifndef SGEMMOTCOPY
105110
SGEMMOTCOPY = ../generic/gemm_tcopy_2.c
111+
endif
112+
ifndef SGEMMONCOPYOBJ
106113
SGEMMONCOPYOBJ = sgemm_oncopy$(TSUFFIX).$(SUFFIX)
114+
endif
115+
ifndef SGEMMOTCOPYOBJ
107116
SGEMMOTCOPYOBJ = sgemm_otcopy$(TSUFFIX).$(SUFFIX)
117+
endif
108118

119+
ifndef DGEMMKERNEL
109120
DGEMMKERNEL = ../generic/gemmkernel_2x2.c
121+
endif
122+
ifndef DGEMMONCOPY
110123
DGEMMONCOPY = ../generic/gemm_ncopy_2.c
124+
endif
125+
ifndef DGEMMOTCOPY
111126
DGEMMOTCOPY = ../generic/gemm_tcopy_2.c
127+
endif
128+
ifndef DGEMMONCOPYOBJ
112129
DGEMMONCOPYOBJ = dgemm_oncopy$(TSUFFIX).$(SUFFIX)
130+
endif
131+
ifndef DGEMMOTCOPYOBJ
113132
DGEMMOTCOPYOBJ = dgemm_otcopy$(TSUFFIX).$(SUFFIX)
133+
endif
114134

115135
CGEMMKERNEL = ../generic/zgemmkernel_2x2.c
116136
CGEMMONCOPY = ../generic/zgemm_ncopy_2.c
@@ -124,15 +144,31 @@ ZGEMMOTCOPY = ../generic/zgemm_tcopy_2.c
124144
ZGEMMONCOPYOBJ = zgemm_oncopy$(TSUFFIX).$(SUFFIX)
125145
ZGEMMOTCOPYOBJ = zgemm_otcopy$(TSUFFIX).$(SUFFIX)
126146

147+
ifndef STRSMKERNEL_LN
127148
STRSMKERNEL_LN = ../generic/trsm_kernel_LN.c
149+
endif
150+
ifndef STRSMKERNEL_LT
128151
STRSMKERNEL_LT = ../generic/trsm_kernel_LT.c
152+
endif
153+
ifndef STRSMKERNEL_RN
129154
STRSMKERNEL_RN = ../generic/trsm_kernel_RN.c
155+
endif
156+
ifndef STRSMKERNEL_RT
130157
STRSMKERNEL_RT = ../generic/trsm_kernel_RT.c
158+
endif
131159

160+
ifndef DTRSMKERNEL_LN
132161
DTRSMKERNEL_LN = ../generic/trsm_kernel_LN.c
162+
endif
163+
ifndef DTRSMKERNEL_LT
133164
DTRSMKERNEL_LT = ../generic/trsm_kernel_LT.c
165+
endif
166+
ifndef DTRSMKERNEL_RN
134167
DTRSMKERNEL_RN = ../generic/trsm_kernel_RN.c
168+
endif
169+
ifndef DTRSMKERNEL_RT
135170
DTRSMKERNEL_RT = ../generic/trsm_kernel_RT.c
171+
endif
136172

137173
CTRSMKERNEL_LN = ../generic/trsm_kernel_LN.c
138174
CTRSMKERNEL_LT = ../generic/trsm_kernel_LT.c

0 commit comments

Comments
 (0)