Skip to content

Commit 23fbce1

Browse files
author
Raghuveer Devulapalli
committed
Use env var to disable/enable avx512 dispatch
1 parent d6e327a commit 23fbce1

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

lib/x86simdsort.cpp

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,20 +7,22 @@
77

88
static int check_cpu_feature_support(std::string_view cpufeature)
99
{
10-
if (cpufeature == "avx512_spr")
10+
const char* disable_avx512 = std::getenv("XSS_DISABLE_AVX512");
11+
12+
if ((cpufeature == "avx512_spr") && (!disable_avx512))
1113
#ifdef __FLT16_MAX__
1214
return __builtin_cpu_supports("avx512f")
1315
&& __builtin_cpu_supports("avx512fp16")
1416
&& __builtin_cpu_supports("avx512vbmi2");
1517
#else
1618
return 0;
1719
#endif
18-
else if (cpufeature == "avx512_icl")
20+
else if ((cpufeature == "avx512_icl") && (!disable_avx512))
1921
return __builtin_cpu_supports("avx512f")
2022
&& __builtin_cpu_supports("avx512vbmi2")
2123
&& __builtin_cpu_supports("avx512bw")
2224
&& __builtin_cpu_supports("avx512vl");
23-
else if (cpufeature == "avx512_skx")
25+
else if ((cpufeature == "avx512_skx") && (!disable_avx512))
2426
return __builtin_cpu_supports("avx512f")
2527
&& __builtin_cpu_supports("avx512dq")
2628
&& __builtin_cpu_supports("avx512vl");

0 commit comments

Comments
 (0)