Skip to content

Commit 1afd724

Browse files
xavierarteagaqarlosalberto
authored andcommitted
support: included CPU features depend on compiler
1 parent 85abf69 commit 1afd724

File tree

1 file changed

+25
-12
lines changed

1 file changed

+25
-12
lines changed

include/srsran/support/cpu_features.h

Lines changed: 25 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -135,28 +135,41 @@ inline bool cpu_supports_feature(cpu_feature feature)
135135

136136
static const std::vector<cpu_feature> cpu_features_included = {
137137
#ifdef __x86_64__
138-
#ifdef HAVE_SSE
138+
#ifdef __SSE4_1__
139139
cpu_feature::sse4_1,
140-
#endif // HAVE_SSE
141-
#ifdef HAVE_AVX
140+
#endif // __SSE4_1__
141+
#ifdef __AVX2__
142142
cpu_feature::avx,
143-
#endif // HAVE_AVX
144-
#ifdef HAVE_AVX2
143+
#endif // __AVX2__
144+
#ifdef __AVX2__
145145
cpu_feature::avx2,
146-
#endif // HAVE_AVX2
147-
#ifdef HAVE_AVX512
146+
#endif // __AVX2__
147+
#ifdef __AVX512F__
148148
cpu_feature::avx512f,
149+
#endif // __AVX512F__
150+
#ifdef __AVX512BW__
149151
cpu_feature::avx512bw,
150-
#endif // HAVE_AVX512
151-
#ifdef HAVE_FMA
152+
#endif // __AVX512BW__
153+
#ifdef __AVX512VL__
154+
cpu_feature::avx512vl,
155+
#endif // __AVX512VL__
156+
#ifdef __AVX512DQ__
157+
cpu_feature::avx512dq,
158+
#endif // __AVX512DQ__
159+
#ifdef __AVX512CD__
160+
cpu_feature::avx512cd,
161+
#endif // __AVX512CD__
162+
#ifdef __FMA__
152163
cpu_feature::fma,
153-
#endif // HAVE_FMA
164+
#endif // __FMA__
165+
#ifdef __PCLMUL__
154166
cpu_feature::pclmul,
167+
#endif // __PCLMUL__
155168
#endif // __x86_64__
156169
#ifdef __aarch64__
157-
#ifdef HAVE_NEON
170+
#ifdef __ARM_NEON
158171
cpu_feature::neon,
159-
#endif // HAVE_NEON
172+
#endif // __ARM_NEON
160173
#endif // __aarch64__
161174
};
162175

0 commit comments

Comments
 (0)