From 22f9b8a6fc169ea84a978175e58964ac5d95f2cb Mon Sep 17 00:00:00 2001 From: Angelos Katharopoulos Date: Thu, 3 Jul 2025 17:33:25 -0700 Subject: [PATCH] More JIT fixes --- mlx/backend/metal/jit/includes.h | 1 + 1 file changed, 1 insertion(+) diff --git a/mlx/backend/metal/jit/includes.h b/mlx/backend/metal/jit/includes.h index 27ae22d05..b380a8374 100644 --- a/mlx/backend/metal/jit/includes.h +++ b/mlx/backend/metal/jit/includes.h @@ -34,6 +34,7 @@ const char* steel_gemm_fused(); const char* steel_gemm_masked(); const char* steel_gemm_splitk(); const char* steel_gemm_gather(); +const char* steel_gemm_segmented(); const char* conv(); const char* steel_conv(); const char* steel_conv_general();