diff --git a/lapack/potrf/potrf_parallel.c b/lapack/potrf/potrf_parallel.c index 2bbbc7bb3e..1e512848b8 100644 --- a/lapack/potrf/potrf_parallel.c +++ b/lapack/potrf/potrf_parallel.c @@ -407,7 +407,7 @@ static int thread_driver(blas_arg_t *args, FLOAT *sa, FLOAT *sb){ mode = BLAS_DOUBLE | BLAS_REAL; mask = MAX(DGEMM_UNROLL_M, DGEMM_UNROLL_N) - 1; #elif defined(BFLOAT16) - mode = BLAS_HALF | BLAS_REAL; + mode = BLAS_BFLOAT16 | BLAS_REAL; mask = MAX(SBGEMM_UNROLL_M, SBGEMM_UNROLL_N) - 1; #else mode = BLAS_SINGLE | BLAS_REAL;