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

format (#2700)

Browse Source
This commit is contained in:
Awni Hannun
2025-10-23 12:24:56 -07:00
parent 5c051678a3
commit 7b34dc366e
2 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
mlx/backend/cuda/quantized/cuda_fp4.h
View File

@@ -47,8 +47,7 @@ struct __nv_fp4_e2m1 {
-2.0f, -2.0f,
-3.0f, -3.0f,
-4.0f, -4.0f,
-6.0f -6.0f};
};
return LUT[__x]; return LUT[__x];
} }

1
mlx/backend/cuda/quantized/fp_quantize.cu
View File

@@ -8,6 +8,7 @@
#include <cooperative_groups.h> #include <cooperative_groups.h>
#include <cooperative_groups/reduce.h> #include <cooperative_groups/reduce.h>
#include <cuda_fp4.h> #include <cuda_fp4.h>
#include <cuda_fp8.h>
namespace mlx::core { namespace mlx::core {
namespace cu { namespace cu {