saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-14 20:48:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add support for checking for F16C support (https://en.wikipedia.org/wiki/F16C). (#212)

Browse source
This commit is contained in:
Jesse Benson 2018-12-19 09:25:25 -08:00 committed by Weixing Zhang
parent ab350fa4c7
commit 0248390e4d
2 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
onnxruntime/core/common/cpuid_info.cc
View file

@ -56,6 +56,7 @@ CPUIDInfo::CPUIDInfo() noexcept {
int value = XGETBV();
bool has_avx = (data[2] & (1 << 28)) && ((value & AVX_MASK) == AVX_MASK);
bool has_avx512 = (value & AVX512_MASK) == AVX512_MASK;
has_f16c_ = has_avx && (data[2] & (1 << 29)) && (data[3] & (1 << 26));
if (num_IDs >= 7) {
GetCPUID(7, data);

2
onnxruntime/core/common/cpuid_info.h
View file

@ -14,11 +14,13 @@ public:
bool HasAVX2() const { return has_avx2_; }
bool HasAVX512f() const { return has_avx512f_; }
bool HasF16C() const { return has_f16c_; }
private:
CPUIDInfo() noexcept;
bool has_avx2_{false};
bool has_avx512f_{false};
bool has_f16c_{false};
};
}