File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -524,7 +524,9 @@ void blas_set_parameter(void){
524524 xgemm_p = ((xgemm_p + XGEMM_UNROLL_M - 1 )/XGEMM_UNROLL_M ) * XGEMM_UNROLL_M ;
525525#endif
526526
527+ #ifdef BUILD_BFLOAT16
527528 sbgemm_r = (((BUFFER_SIZE - ((SBGEMM_P * SBGEMM_Q * 4 + GEMM_OFFSET_A + GEMM_ALIGN ) & ~GEMM_ALIGN )) / (SBGEMM_Q * 4 )) - 15 ) & ~15 ;
529+ #endif
528530 sgemm_r = (((BUFFER_SIZE - ((SGEMM_P * SGEMM_Q * 4 + GEMM_OFFSET_A + GEMM_ALIGN ) & ~GEMM_ALIGN )) / (SGEMM_Q * 4 )) - 15 ) & ~15 ;
529531 dgemm_r = (((BUFFER_SIZE - ((DGEMM_P * DGEMM_Q * 8 + GEMM_OFFSET_A + GEMM_ALIGN ) & ~GEMM_ALIGN )) / (DGEMM_Q * 8 )) - 15 ) & ~15 ;
530532 cgemm_r = (((BUFFER_SIZE - ((CGEMM_P * CGEMM_Q * 8 + GEMM_OFFSET_A + GEMM_ALIGN ) & ~GEMM_ALIGN )) / (CGEMM_Q * 8 )) - 15 ) & ~15 ;
@@ -630,7 +632,9 @@ void blas_set_parameter(void){
630632 xgemm_p = 16 * (size + 1 );
631633#endif
632634
635+ #ifdef BUILD_BFLOAT16
633636 sbgemm_r = (((BUFFER_SIZE - ((SBGEMM_P * SBGEMM_Q * 4 + GEMM_OFFSET_A + GEMM_ALIGN ) & ~GEMM_ALIGN )) / (SBGEMM_Q * 4 )) - 15 ) & ~15 ;
637+ #endif
634638 sgemm_r = (((BUFFER_SIZE - ((SGEMM_P * SGEMM_Q * 4 + GEMM_OFFSET_A + GEMM_ALIGN ) & ~GEMM_ALIGN )) / (SGEMM_Q * 4 )) - 15 ) & ~15 ;
635639 dgemm_r = (((BUFFER_SIZE - ((DGEMM_P * DGEMM_Q * 8 + GEMM_OFFSET_A + GEMM_ALIGN ) & ~GEMM_ALIGN )) / (DGEMM_Q * 8 )) - 15 ) & ~15 ;
636640 cgemm_r = (((BUFFER_SIZE - ((CGEMM_P * CGEMM_Q * 8 + GEMM_OFFSET_A + GEMM_ALIGN ) & ~GEMM_ALIGN )) / (CGEMM_Q * 8 )) - 15 ) & ~15 ;
You can’t perform that action at this time.
0 commit comments