GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-09-13 09:53:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

Disable float16 packet casting if native AVX512 f16 is available.

Browse Source
This commit is contained in:
Antonio Sánchez 2024-02-14 20:05:00 +00:00
parent 5361dea833
commit a24bf2e9a2
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Eigen/src/Core/arch/AVX512/TypeCasting.h
View File

@ -37,10 +37,12 @@ struct type_casting_traits<double, int> : vectorized_type_casting_traits<double,
template <> template <>
struct type_casting_traits<int, double> : vectorized_type_casting_traits<int, double> {}; struct type_casting_traits<int, double> : vectorized_type_casting_traits<int, double> {};
#ifndef EIGEN_VECTORIZE_AVX512FP16
template <> template <>
struct type_casting_traits<half, float> : vectorized_type_casting_traits<half, float> {}; struct type_casting_traits<half, float> : vectorized_type_casting_traits<half, float> {};
template <> template <>
struct type_casting_traits<float, half> : vectorized_type_casting_traits<float, half> {}; struct type_casting_traits<float, half> : vectorized_type_casting_traits<float, half> {};
#endif
template <> template <>
struct type_casting_traits<bfloat16, float> : vectorized_type_casting_traits<bfloat16, float> {}; struct type_casting_traits<bfloat16, float> : vectorized_type_casting_traits<bfloat16, float> {};