Skip to content

Commit 2d6cd47

Browse files
authored
Masked gemv (#1211)
1 parent fe3167d commit 2d6cd47

File tree

6 files changed

+1627
-319
lines changed

6 files changed

+1627
-319
lines changed

mlx/backend/metal/kernels/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@ endfunction(build_kernel)
3737
build_kernel(arg_reduce)
3838
build_kernel(conv steel/conv/params.h)
3939
build_kernel(gemv steel/utils.h)
40+
build_kernel(gemv_masked steel/utils.h)
4041
build_kernel(layer_norm)
4142
build_kernel(random)
4243
build_kernel(rms_norm)

0 commit comments

Comments
 (0)