search

openxla / xla

A machine learning compiler for GPUs, CPUs, and ML accelerators
Apache License 2.0
2.4k stars 361 forks source link

[XLA:CPU] Fix the detection of FP16 extension to AVX512 #14368

Open akhilgoe opened 3 days ago

akhilgoe commented 3 days ago

Currently, the check for AVX512_FP16 is not correct. This PR fixes it to match xbyak code.