mirror of
https://github.com/ml-explore/mlx.git
synced 2025-12-16 01:49:05 +08:00
format (#2700)
This commit is contained in:
@@ -47,8 +47,7 @@ struct __nv_fp4_e2m1 {
|
||||
-2.0f,
|
||||
-3.0f,
|
||||
-4.0f,
|
||||
-6.0f
|
||||
};
|
||||
-6.0f};
|
||||
|
||||
return LUT[__x];
|
||||
}
|
||||
|
||||
@@ -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 {
|
||||
|
||||
Reference in New Issue
Block a user