floraachy
/
OpenBLAS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use auvx to detect CPUCFG on mips/loongson 

It's safer and easier than SIGILL.

Signed-off-by: Jiaxun Yang <jiaxun.yang@flygoat.com>
This commit is contained in:
Jiaxun Yang 2022-07-31 18:17:07 +01:00
parent ef9c976a94
commit 19d4f90c44
1 changed files with 21 additions and 59 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

80
driver/others/dynamic_mips64.c
View File

@ -38,6 +38,15 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#include <sys/resource.h> #include <sys/resource.h>
#include "common.h" #include "common.h"
#if (defined OS_LINUX || defined OS_ANDROID)
#include <asm/hwcap.h>
#include <sys/auxv.h>
#ifndef HWCAP_LOONGSON_CPUCFG
#define HWCAP_LOONGSON_CPUCFG (1 << 14)
#endif
#endif
extern gotoblas_t gotoblas_LOONGSON3R3; extern gotoblas_t gotoblas_LOONGSON3R3;
extern gotoblas_t gotoblas_LOONGSON3R4; extern gotoblas_t gotoblas_LOONGSON3R4;
@ -81,59 +90,10 @@ static gotoblas_t *force_coretype(char *coretype) {
return NULL; return NULL;
} }
#if (defined OS_LINUX || defined OS_ANDROID)
#define MMI_MASK 0x00000010 #define MMI_MASK 0x00000010
#define MSA_MASK 0x00000020 #define MSA_MASK 0x00000020
int fd[2];
int support_cpucfg;
static void handler(int signum)
{
close(fd[1]);
exit(1);
}
/* Brief : Function to check if cpucfg supported on loongson
* Return: 1 supported
* 0 not supported
*/
static int cpucfg_test(void) {
pid_t pid;
int status = 0;
support_cpucfg = 0;
pipe(fd);
pid = fork();
if (pid == 0) { /* Subprocess */
struct sigaction act;
close(fd[0]);
/* Set signal action for SIGILL. */
act.sa_handler = handler;
sigaction(SIGILL,&act,NULL);
/* Execute cpucfg in subprocess. */
__asm__ volatile(
".insn \n\t"
".word (0xc8080118) \n\t"
:::
);
support_cpucfg = 1;
write(fd[1],&support_cpucfg,sizeof(support_cpucfg));
close(fd[1]);
exit(0);
} else if (pid > 0){ /* Parent process*/
close(fd[1]);
if ((waitpid(pid,&status,0) <= 0) ||
(read(fd[0],&support_cpucfg,sizeof(support_cpucfg)) <= 0))
support_cpucfg = 0;
close(fd[0]);
} else {
support_cpucfg = 0;
}
return support_cpucfg;
}
static gotoblas_t *get_coretype_from_cpucfg(void) { static gotoblas_t *get_coretype_from_cpucfg(void) {
int flag = 0; int flag = 0;
__asm__ volatile( __asm__ volatile(
@ -153,7 +113,7 @@ static gotoblas_t *get_coretype_from_cpucfg(void) {
} }
static gotoblas_t *get_coretype_from_cpuinfo(void) { static gotoblas_t *get_coretype_from_cpuinfo(void) {
#ifdef linux #ifdef __linux
FILE *infile; FILE *infile;
char buffer[512], *p; char buffer[512], *p;
@ -176,17 +136,19 @@ static gotoblas_t *get_coretype_from_cpuinfo(void) {
return NULL; return NULL;
} }
#endif #endif
return NULL; return NULL;
} }
#endif
static gotoblas_t *get_coretype(void) { static gotoblas_t *get_coretype(void) {
int ret = 0; #if (!defined OS_LINUX && !defined OS_ANDROID)
return NULL;
ret = cpucfg_test(); #else
if (ret == 1) if (!(getauxval(AT_HWCAP) & HWCAP_LOONGSON_CPUCFG))
return get_coretype_from_cpucfg(); return get_coretype_from_cpucfg();
else else
return get_coretype_from_cpuinfo(); return get_coretype_from_cpuinfo();
#endif
} }
void gotoblas_dynamic_init(void) { void gotoblas_dynamic_init(void) {