File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 4949GOTO_LAPACK_TARGETS =
5050endif
5151
52- ifeq ($(BUILD_HALF ) ,1)
53- GOTO_HALF_TARGETS =shgemm .goto
52+ ifeq ($(BUILD_BFLOAT16 ) ,1)
53+ GOTO_HALF_TARGETS =sbgemm .goto
5454else
5555GOTO_HALF_TARGETS =
5656endif
@@ -620,8 +620,8 @@ zcholesky.essl : zcholesky.$(SUFFIX)
620620 -$(CC ) $(CFLAGS ) -o $(@F ) $^ $(LIBESSL ) $(CEXTRALIB ) $(EXTRALIB ) $(FEXTRALIB )
621621
622622# #################################### Sgemm ####################################################
623- ifeq ($(BUILD_HALF ) ,1)
624- shgemm .goto : shgemm .$(SUFFIX ) ../$(LIBNAME )
623+ ifeq ($(BUILD_BFLOAT16 ) ,1)
624+ sbgemm .goto : sbgemm .$(SUFFIX ) ../$(LIBNAME )
625625 $(CC ) $(CFLAGS ) -o $(@F ) $^ $(CEXTRALIB ) $(EXTRALIB ) $(FEXTRALIB ) -lm
626626endif
627627
@@ -2927,8 +2927,8 @@ ccholesky.$(SUFFIX) : cholesky.c
29272927zcholesky.$(SUFFIX ) : cholesky.c
29282928 $(CC ) $(CFLAGS ) -c -DCOMPLEX -DDOUBLE -o $(@F ) $^
29292929
2930- ifeq ($(BUILD_HALF ) ,1)
2931- shgemm .$(SUFFIX ) : gemm.c
2930+ ifeq ($(BUILD_BFLOAT16 ) ,1)
2931+ sbgemm .$(SUFFIX ) : gemm.c
29322932 $(CC ) $(CFLAGS ) -c -DHALF -UCOMPLEX -UDOUBLE -o $(@F ) $^
29332933endif
29342934
Original file line number Diff line number Diff line change @@ -40,7 +40,7 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
4040#ifdef DOUBLE
4141#define GEMM BLASFUNC(dgemm)
4242#elif defined(HALF )
43- #define GEMM BLASFUNC(shgemm )
43+ #define GEMM BLASFUNC(sbgemm )
4444#else
4545#define GEMM BLASFUNC(sgemm)
4646#endif
You can’t perform that action at this time.
0 commit comments