Browse Source

Merge pull request #5364 from martin-frbg/blashalf

change BLAS_HALF to BLAS_BFLOAT16 in parallelized POTRF (another missed rename)
pull/5365/head
Martin Kroeker GitHub 2 months ago
parent
commit
15d6e58510
No known key found for this signature in database GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      lapack/potrf/potrf_parallel.c

+ 1
- 1
lapack/potrf/potrf_parallel.c View File

@@ -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;


Loading…
Cancel
Save