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 METAL quantization in JIT (#2553)

Browse Source
This commit is contained in:
Awni Hannun
2025-08-28 18:26:25 -07:00
committed by GitHub
parent d363a76aa4
commit 827003d568
9 changed files with 154 additions and 89 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
mlx/backend/metal/CMakeLists.txt
View File

@@ -77,7 +77,10 @@ if(MLX_METAL_JIT)
make_jit_source(steel/conv/kernels/steel_conv)
make_jit_source(steel/conv/kernels/steel_conv_general kernels/steel/defines.h
kernels/steel/conv/loaders/loader_general.h)
make_jit_source(quantized)
make_jit_source(quantized_utils)
make_jit_source(quantized kernels/quantized_utils.h)
make_jit_source(fp4_quantized kernels/quantized_utils.h)
make_jit_source(gemv_masked)
else()
target_sources(mlx PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/nojit_kernels.cpp)