floraachy
/
OpenBLAS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Change "HALF" and "sh" to "BFLOAT16" and "sb"

This commit is contained in:
Martin Kroeker 2020-10-12 00:08:29 +02:00 committed by GitHub
parent e3a29f6b58
commit 7ae9e8960e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
benchmark/Makefile
View File

@ -49,8 +49,8 @@ else
GOTO_LAPACK_TARGETS= GOTO_LAPACK_TARGETS=
endif endif
ifeq ($(BUILD_HALF),1) ifeq ($(BUILD_BFLOAT16),1)
GOTO_HALF_TARGETS=shgemm.goto GOTO_HALF_TARGETS=sbgemm.goto
else else
GOTO_HALF_TARGETS= GOTO_HALF_TARGETS=
endif endif
@ -620,8 +620,8 @@ zcholesky.essl : zcholesky.$(SUFFIX)
-$(CC) $(CFLAGS) -o $(@F) $^ $(LIBESSL) $(CEXTRALIB) $(EXTRALIB) $(FEXTRALIB) -$(CC) $(CFLAGS) -o $(@F) $^ $(LIBESSL) $(CEXTRALIB) $(EXTRALIB) $(FEXTRALIB)
##################################### Sgemm #################################################### ##################################### Sgemm ####################################################
ifeq ($(BUILD_HALF),1) ifeq ($(BUILD_BFLOAT16),1)
shgemm.goto : shgemm.$(SUFFIX) ../$(LIBNAME) sbgemm.goto : sbgemm.$(SUFFIX) ../$(LIBNAME)
$(CC) $(CFLAGS) -o $(@F) $^ $(CEXTRALIB) $(EXTRALIB) $(FEXTRALIB) -lm $(CC) $(CFLAGS) -o $(@F) $^ $(CEXTRALIB) $(EXTRALIB) $(FEXTRALIB) -lm
endif endif
@ -2927,8 +2927,8 @@ ccholesky.$(SUFFIX) : cholesky.c
zcholesky.$(SUFFIX) : cholesky.c zcholesky.$(SUFFIX) : cholesky.c
$(CC) $(CFLAGS) -c -DCOMPLEX -DDOUBLE -o $(@F) $^ $(CC) $(CFLAGS) -c -DCOMPLEX -DDOUBLE -o $(@F) $^
ifeq ($(BUILD_HALF),1) ifeq ($(BUILD_BFLOAT16),1)
shgemm.$(SUFFIX) : gemm.c sbgemm.$(SUFFIX) : gemm.c
$(CC) $(CFLAGS) -c -DHALF -UCOMPLEX -UDOUBLE -o $(@F) $^ $(CC) $(CFLAGS) -c -DHALF -UCOMPLEX -UDOUBLE -o $(@F) $^
endif endif

2
benchmark/gemm.c
View File

@ -40,7 +40,7 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#ifdef DOUBLE #ifdef DOUBLE
#define GEMM BLASFUNC(dgemm) #define GEMM BLASFUNC(dgemm)
#elif defined(HALF) #elif defined(HALF)
#define GEMM BLASFUNC(shgemm) #define GEMM BLASFUNC(sbgemm)
#else #else
#define GEMM BLASFUNC(sgemm) #define GEMM BLASFUNC(sgemm)
#endif #endif