zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix for older OS

Browse Source
This commit is contained in:
Awni Hannun
2025-10-20 14:23:09 -07:00
parent 6461fcb4aa
commit 7b329a816e
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mlx/backend/cpu/unary_ops.h
View File

@@ -122,7 +122,9 @@ struct ToFP8 {
Simd<uint8_t, N> operator()(Simd<T, N> f) {
uint32_t fp8_max = 1087 << 20;
auto denorm_mask = Simd<uint32_t, N>(141 << 23);
auto f_bits = fp32_to_bits(Simd<float, N>(f));
Simd<uint32_t, N> f_bits;
Simd<float, N> f32 = f;
f_bits = fp32_to_bits(f32);
Simd<uint8_t, N> result = 0u;
auto sign = f_bits & 0x80000000;
f_bits = f_bits ^ sign;