Skip to content

Commit 0cf656f

Browse files
martin-frbgmartin-frbg
authored
Add copies of GEMMT under its new name GEMMTR
1 parent f66e6d3 commit 0cf656f

File tree

2 files changed

+43
-11
lines changed

2 files changed

+43
-11
lines changed

‎interface/CMakeLists.txt‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -107,6 +107,9 @@ endif ()
107107

108108
# trmm is trsm with a compiler flag set
109109
GenerateNamedObjects("trsm.c" "TRMM" "trmm" ${CBLAS_FLAG})
110+
111+
# gemmtr is gemmt under the name adopted by the Reference BLAS
112+
GenerateNamedObjects("gemm.c" "" "gemmtr" ${CBLAS_FLAG})
110113

111114
# max and imax are compiled 4 times
112115
GenerateNamedObjects("max.c" "" "" ${CBLAS_FLAG})
@@ -123,6 +126,7 @@ if (BUILD_BFLOAT16)
123126
GenerateNamedObjects("bf16dot.c" "" "sbdot" ${CBLAS_FLAG} "" "" true "BFLOAT16")
124127
GenerateNamedObjects("gemm.c" "" "sbgemm" ${CBLAS_FLAG} "" "" true "BFLOAT16")
125128
GenerateNamedObjects("gemmt.c" "" "sbgemmt" ${CBLAS_FLAG} "" "" true "BFLOAT16")
129+
GenerateNamedObjects("gemmt.c" "" "sbgemmtr" ${CBLAS_FLAG} "" "" true "BFLOAT16")
126130
GenerateNamedObjects("sbgemv.c" "" "sbgemv" ${CBLAS_FLAG} "" "" true "BFLOAT16")
127131
GenerateNamedObjects("tobf16.c" "SINGLE_PREC" "sbstobf16" ${CBLAS_FLAG} "" "" true "BFLOAT16")
128132
GenerateNamedObjects("tobf16.c" "DOUBLE_PREC" "sbdtobf16" ${CBLAS_FLAG} "" "" true "BFLOAT16")

‎interface/Makefile‎

Lines changed: 39 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -44,12 +44,12 @@ SBLAS3OBJS = \
4444
sgemm.$(SUFFIX) ssymm.$(SUFFIX) strmm.$(SUFFIX) \
4545
strsm.$(SUFFIX) ssyrk.$(SUFFIX) ssyr2k.$(SUFFIX) \
4646
somatcopy.$(SUFFIX) simatcopy.$(SUFFIX)\
47-
sgeadd.$(SUFFIX) sgemmt.$(SUFFIX)
47+
sgeadd.$(SUFFIX) sgemmt.$(SUFFIX) sgemmtr.$(SUFFIX)
4848

4949
ifeq ($(BUILD_BFLOAT16),1)
5050
SBBLAS1OBJS = sbdot.$(SUFFIX)
5151
SBBLAS2OBJS = sbgemv.$(SUFFIX)
52-
SBBLAS3OBJS = sbgemm.$(SUFFIX) sbgemmt.$(SUFFIX)
52+
SBBLAS3OBJS = sbgemm.$(SUFFIX) sbgemmt.$(SUFFIX) sbgemmtr.$(SUFFIX)
5353
SBEXTOBJS = sbstobf16.$(SUFFIX) sbdtobf16.$(SUFFIX) sbf16tos.$(SUFFIX) dbf16tod.$(SUFFIX)
5454
endif
5555

@@ -76,7 +76,7 @@ DBLAS3OBJS = \
7676
dgemm.$(SUFFIX) dsymm.$(SUFFIX) dtrmm.$(SUFFIX) \
7777
dtrsm.$(SUFFIX) dsyrk.$(SUFFIX) dsyr2k.$(SUFFIX) \
7878
domatcopy.$(SUFFIX) dimatcopy.$(SUFFIX)\
79-
dgeadd.$(SUFFIX) dgemmt.$(SUFFIX)
79+
dgeadd.$(SUFFIX) dgemmt.$(SUFFIX) dgemmtr.$(SUFFIX)
8080

8181
CBLAS1OBJS = \
8282
caxpy.$(SUFFIX) caxpyc.$(SUFFIX) cswap.$(SUFFIX) \
@@ -105,7 +105,7 @@ CBLAS3OBJS = \
105105
ctrsm.$(SUFFIX) csyrk.$(SUFFIX) csyr2k.$(SUFFIX) \
106106
chemm.$(SUFFIX) cherk.$(SUFFIX) cher2k.$(SUFFIX) \
107107
comatcopy.$(SUFFIX) cimatcopy.$(SUFFIX)\
108-
cgeadd.$(SUFFIX) cgemmt.$(SUFFIX)
108+
cgeadd.$(SUFFIX) cgemmt.$(SUFFIX) cgemmtr.$(SUFFIX)
109109

110110
ZBLAS1OBJS = \
111111
zaxpy.$(SUFFIX) zaxpyc.$(SUFFIX) zswap.$(SUFFIX) \
@@ -134,7 +134,7 @@ ZBLAS3OBJS = \
134134
ztrsm.$(SUFFIX) zsyrk.$(SUFFIX) zsyr2k.$(SUFFIX) \
135135
zhemm.$(SUFFIX) zherk.$(SUFFIX) zher2k.$(SUFFIX) \
136136
zomatcopy.$(SUFFIX) zimatcopy.$(SUFFIX)\
137-
zgeadd.$(SUFFIX) zgemmt.$(SUFFIX)
137+
zgeadd.$(SUFFIX) zgemmt.$(SUFFIX) zgemmtr.$(SUFFIX)
138138

139139
ifeq ($(SUPPORT_GEMM3M), 1)
140140

@@ -282,12 +282,12 @@ CSBLAS2OBJS = \
282282
CSBLAS3OBJS = \
283283
cblas_sgemm.$(SUFFIX) cblas_ssymm.$(SUFFIX) cblas_strmm.$(SUFFIX) cblas_strsm.$(SUFFIX) \
284284
cblas_ssyrk.$(SUFFIX) cblas_ssyr2k.$(SUFFIX) cblas_somatcopy.$(SUFFIX) cblas_simatcopy.$(SUFFIX)\
285-
cblas_sgeadd.$(SUFFIX) cblas_sgemmt.$(SUFFIX) cblas_sgemm_batch.$(SUFFIX)
285+
cblas_sgeadd.$(SUFFIX) cblas_sgemmt.$(SUFFIX) cblas_sgemmtr.$(SUFFIX) cblas_sgemm_batch.$(SUFFIX)
286286

287287
ifeq ($(BUILD_BFLOAT16),1)
288288
CSBBLAS1OBJS = cblas_sbdot.$(SUFFIX)
289289
CSBBLAS2OBJS = cblas_sbgemv.$(SUFFIX)
290-
CSBBLAS3OBJS = cblas_sbgemm.$(SUFFIX) cblas_sbgemmt.$(SUFFIX) cblas_sbgemm_batch.$(SUFFIX)
290+
CSBBLAS3OBJS = cblas_sbgemm.$(SUFFIX) cblas_sbgemmt.$(SUFFIX) cblas_sbgemmtr.$(SUFFIX) cblas_sbgemm_batch.$(SUFFIX)
291291
CSBEXTOBJS = cblas_sbstobf16.$(SUFFIX) cblas_sbdtobf16.$(SUFFIX) cblas_sbf16tos.$(SUFFIX) cblas_dbf16tod.$(SUFFIX)
292292
endif
293293

@@ -308,7 +308,7 @@ CDBLAS2OBJS = \
308308
CDBLAS3OBJS += \
309309
cblas_dgemm.$(SUFFIX) cblas_dsymm.$(SUFFIX) cblas_dtrmm.$(SUFFIX) cblas_dtrsm.$(SUFFIX) \
310310
cblas_dsyrk.$(SUFFIX) cblas_dsyr2k.$(SUFFIX) cblas_domatcopy.$(SUFFIX) cblas_dimatcopy.$(SUFFIX) \
311-
cblas_dgeadd.$(SUFFIX) cblas_dgemmt.$(SUFFIX) cblas_dgemm_batch.$(SUFFIX)
311+
cblas_dgeadd.$(SUFFIX) cblas_dgemmt.$(SUFFIX) cblas_dgemmtr.$(SUFFIX) cblas_dgemm_batch.$(SUFFIX)
312312

313313
CCBLAS1OBJS = \
314314
cblas_icamax.$(SUFFIX) cblas_icamin.$(SUFFIX) cblas_scasum.$(SUFFIX) cblas_caxpy.$(SUFFIX) \
@@ -333,7 +333,7 @@ CCBLAS3OBJS = \
333333
cblas_csyrk.$(SUFFIX) cblas_csyr2k.$(SUFFIX) \
334334
cblas_chemm.$(SUFFIX) cblas_cherk.$(SUFFIX) cblas_cher2k.$(SUFFIX) \
335335
cblas_comatcopy.$(SUFFIX) cblas_cimatcopy.$(SUFFIX)\
336-
cblas_cgeadd.$(SUFFIX) cblas_cgemmt.$(SUFFIX) cblas_cgemm_batch.$(SUFFIX)
336+
cblas_cgeadd.$(SUFFIX) cblas_cgemmt.$(SUFFIX) cblas_cgemmtr.$(SUFFIX) cblas_cgemm_batch.$(SUFFIX)
337337

338338
CXERBLAOBJ = \
339339
cblas_xerbla.$(SUFFIX)
@@ -364,7 +364,7 @@ CZBLAS3OBJS = \
364364
cblas_zsyrk.$(SUFFIX) cblas_zsyr2k.$(SUFFIX) \
365365
cblas_zhemm.$(SUFFIX) cblas_zherk.$(SUFFIX) cblas_zher2k.$(SUFFIX)\
366366
cblas_zomatcopy.$(SUFFIX) cblas_zimatcopy.$(SUFFIX) \
367-
cblas_zgeadd.$(SUFFIX) cblas_zgemmt.$(SUFFIX) cblas_zgemm_batch.$(SUFFIX)
367+
cblas_zgeadd.$(SUFFIX) cblas_zgemmt.$(SUFFIX) cblas_zgemmtr.$(SUFFIX) cblas_zgemm_batch.$(SUFFIX)
368368

369369

370370
ifeq ($(SUPPORT_GEMM3M), 1)
@@ -1305,6 +1305,8 @@ sbgemm.$(SUFFIX) sbgemm.$(PSUFFIX) : gemm.c ../param.h
13051305
$(CC) -c $(CFLAGS) $< -o $(@F)
13061306
sbgemmt.$(SUFFIX) sbgemmt.$(PSUFFIX) : sbgemmt.c ../param.h
13071307
$(CC) -c $(CFLAGS) $< -o $(@F)
1308+
sbgemmtr.$(SUFFIX) sbgemmtr.$(PSUFFIX) : sbgemmt.c ../param.h
1309+
$(CC) -c $(CFLAGS) $< -o $(@F)
13081310
endif
13091311

13101312
sgemm.$(SUFFIX) sgemm.$(PSUFFIX) : gemm.c ../param.h
@@ -1340,7 +1342,19 @@ cgemmt.$(SUFFIX) cgemmt.$(PSUFFIX) : gemmt.c ../param.h
13401342
zgemmt.$(SUFFIX) zgemmt.$(PSUFFIX) : gemmt.c ../param.h
13411343
$(CC) -c $(CFLAGS) $< -o $(@F)
13421344

1343-
xgemmt.$(SUFFIX) xgemmt.$(PSUFFIX) : gemmt.c ../param.h
1345+
sgemmtr.$(SUFFIX) sgemmtr.$(PSUFFIX) : gemmt.c ../param.h
1346+
$(CC) -c $(CFLAGS) $< -o $(@F)
1347+
1348+
dgemmtr.$(SUFFIX) dgemmtr.$(PSUFFIX) : gemmt.c ../param.h
1349+
$(CC) -c $(CFLAGS) $< -o $(@F)
1350+
1351+
qgemmtr.$(SUFFIX) qgemmtr.$(PSUFFIX) : gemmt.c ../param.h
1352+
$(CC) -c $(CFLAGS) $< -o $(@F)
1353+
1354+
cgemmtr.$(SUFFIX) cgemmtr.$(PSUFFIX) : gemmt.c ../param.h
1355+
$(CC) -c $(CFLAGS) $< -o $(@F)
1356+
1357+
zgemmtr.$(SUFFIX) zgemmtr.$(PSUFFIX) : gemmt.c ../param.h
13441358
$(CC) -c $(CFLAGS) $< -o $(@F)
13451359

13461360
ssymm.$(SUFFIX) ssymm.$(PSUFFIX) : symm.c
@@ -1966,9 +1980,14 @@ cblas_zgemm.$(SUFFIX) cblas_zgemm.$(PSUFFIX) : gemm.c ../param.h
19661980
cblas_sgemmt.$(SUFFIX) cblas_sgemmt.$(PSUFFIX) : gemmt.c ../param.h
19671981
$(CC) -DCBLAS -c $(CFLAGS) $< -o $(@F)
19681982

1983+
cblas_sgemmtr.$(SUFFIX) cblas_sgemmtr.$(PSUFFIX) : gemmt.c ../param.h
1984+
$(CC) -DCBLAS -c $(CFLAGS) $< -o $(@F)
1985+
19691986
ifeq ($(BUILD_BFLOAT16),1)
19701987
cblas_sbgemmt.$(SUFFIX) cblas_sbgemmt.$(PSUFFIX) : sbgemmt.c ../param.h
19711988
$(CC) -DCBLAS -c $(CFLAGS) $< -o $(@F)
1989+
cblas_sbgemmtr.$(SUFFIX) cblas_sbgemmtr.$(PSUFFIX) : sbgemmt.c ../param.h
1990+
$(CC) -DCBLAS -c $(CFLAGS) $< -o $(@F)
19721991
endif
19731992

19741993
cblas_dgemmt.$(SUFFIX) cblas_dgemmt.$(PSUFFIX) : gemmt.c ../param.h
@@ -1980,6 +1999,15 @@ cblas_cgemmt.$(SUFFIX) cblas_cgemmt.$(PSUFFIX) : gemmt.c ../param.h
19801999
cblas_zgemmt.$(SUFFIX) cblas_zgemmt.$(PSUFFIX) : gemmt.c ../param.h
19812000
$(CC) -DCBLAS -c $(CFLAGS) $< -o $(@F)
19822001

2002+
cblas_dgemmtr.$(SUFFIX) cblas_dgemmtr.$(PSUFFIX) : gemmt.c ../param.h
2003+
$(CC) -DCBLAS -c $(CFLAGS) $< -o $(@F)
2004+
2005+
cblas_cgemmtr.$(SUFFIX) cblas_cgemmtr.$(PSUFFIX) : gemmt.c ../param.h
2006+
$(CC) -DCBLAS -c $(CFLAGS) $< -o $(@F)
2007+
2008+
cblas_zgemmtr.$(SUFFIX) cblas_zgemmtr.$(PSUFFIX) : gemmt.c ../param.h
2009+
$(CC) -DCBLAS -c $(CFLAGS) $< -o $(@F)
2010+
19832011
cblas_ssymm.$(SUFFIX) cblas_ssymm.$(PSUFFIX) : symm.c
19842012
$(CC) -DCBLAS -c $(CFLAGS) $< -o $(@F)
19852013

0 commit comments

Comments
 (0)