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

View File

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

View File

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