Skip to content

Commit 22b487b

Browse files
authored
Merge pull request #4488 from martin-frbg/issue4475-2
Separate the interface for SBGEMMT from GEMMT
2 parents 818bf30 + fb99fc2 commit 22b487b

File tree

5 files changed

+466
-11
lines changed

5 files changed

+466
-11
lines changed

interface/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,7 @@ endif ()
119119
if (BUILD_BFLOAT16)
120120
GenerateNamedObjects("bf16dot.c" "" "sbdot" ${CBLAS_FLAG} "" "" true "BFLOAT16")
121121
GenerateNamedObjects("gemm.c" "" "sbgemm" ${CBLAS_FLAG} "" "" true "BFLOAT16")
122+
GenerateNamedObjects("gemmt.c" "" "sbgemmt" ${CBLAS_FLAG} "" "" true "BFLOAT16")
122123
GenerateNamedObjects("sbgemv.c" "" "sbgemv" ${CBLAS_FLAG} "" "" true "BFLOAT16")
123124
GenerateNamedObjects("tobf16.c" "SINGLE_PREC" "sbstobf16" ${CBLAS_FLAG} "" "" true "BFLOAT16")
124125
GenerateNamedObjects("tobf16.c" "DOUBLE_PREC" "sbdtobf16" ${CBLAS_FLAG} "" "" true "BFLOAT16")

interface/Makefile

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1303,7 +1303,7 @@ xhpr2.$(SUFFIX) xhpr2.$(PSUFFIX) : zhpr2.c
13031303
ifeq ($(BUILD_BFLOAT16),1)
13041304
sbgemm.$(SUFFIX) sbgemm.$(PSUFFIX) : gemm.c ../param.h
13051305
$(CC) -c $(CFLAGS) $< -o $(@F)
1306-
sbgemmt.$(SUFFIX) sbgemmt.$(PSUFFIX) : gemmt.c ../param.h
1306+
sbgemmt.$(SUFFIX) sbgemmt.$(PSUFFIX) : sbgemmt.c ../param.h
13071307
$(CC) -c $(CFLAGS) $< -o $(@F)
13081308
endif
13091309

@@ -1662,10 +1662,6 @@ cblas_zaxpyc.$(SUFFIX) cblas_zaxpyc.$(PSUFFIX) : zaxpy.c
16621662
cblas_xaxpyc.$(SUFFIX) cblas_xaxpyc.$(PSUFFIX) : zaxpy.c
16631663
$(CC) $(CFLAGS) -DCBLAS -c -DCONJ $< -o $(@F)
16641664

1665-
sscal.$(SUFFIX) sscal.$(PSUFFIX) : scal.c
1666-
$(CC) $(CFLAGS) -c $< -o $(@F)
1667-
1668-
dscal.$(SUFFIX) dscal.$(PSUFFIX) : scal.c
16691665
cblas_zaxpy.$(SUFFIX) cblas_zaxpy.$(PSUFFIX) : zaxpy.c
16701666
$(CC) $(CFLAGS) -DCBLAS -c $< -o $(@F)
16711667

@@ -1971,7 +1967,7 @@ cblas_sgemmt.$(SUFFIX) cblas_sgemmt.$(PSUFFIX) : gemmt.c ../param.h
19711967
$(CC) -DCBLAS -c $(CFLAGS) $< -o $(@F)
19721968

19731969
ifeq ($(BUILD_BFLOAT16),1)
1974-
cblas_sbgemmt.$(SUFFIX) cblas_sbgemmt.$(PSUFFIX) : gemmt.c ../param.h
1970+
cblas_sbgemmt.$(SUFFIX) cblas_sbgemmt.$(PSUFFIX) : sbgemmt.c ../param.h
19751971
$(CC) -DCBLAS -c $(CFLAGS) $< -o $(@F)
19761972
endif
19771973

interface/gemmt.c

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,8 @@ void NAME(char *UPLO, char *TRANSA, char *TRANSB,
158158
uplo = 0;
159159
if (Uplo == 'L')
160160
uplo = 1;
161-
161+
162+
nrowa = m;
162163
if (transa & 1) nrowa = k;
163164
nrowb = k;
164165
#if defined(COMPLEX)

0 commit comments

Comments
 (0)