Skip to content

Commit ed8b377

Browse files
authored
Merge pull request #3628 from martin-frbg/issue3620
DYNAMIC_ARCH: Improve mapping for future AMD cpus
2 parents dbdb686 + 2c62096 commit ed8b377

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

driver/others/dynamic.c

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -855,15 +855,19 @@ static gotoblas_t *get_coretype(void){
855855
openblas_warning(FALLBACK_VERBOSE, BARCELONA_FALLBACK);
856856
return &gotoblas_BARCELONA; //OS doesn't support AVX. Use old kernels.
857857
}
858-
} else if (exfamily == 10) {
858+
} else if (exfamily == 10) {
859+
if(support_avx512_bf16())
860+
return &gotoblas_COOPERLAKE;
861+
if(support_avx512())
862+
return &gotoblas_SKYLAKEX;
859863
if(support_avx())
860864
return &gotoblas_ZEN;
861865
else{
862866
openblas_warning(FALLBACK_VERBOSE, BARCELONA_FALLBACK);
863867
return &gotoblas_BARCELONA; //OS doesn't support AVX. Use old kernels.
864868
}
865869
}else {
866-
return &gotoblas_BARCELONA;
870+
return NULL;
867871
}
868872

869873
}

0 commit comments

Comments
 (0)